MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / SetMklTensor

Method SetMklTensor

tensorflow/core/util/mkl_util.h:314–316  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

312
313 inline const bool IsMklTensor() const { return data_.is_mkl_tensor_; }
314 inline void SetMklTensor(bool is_mkl_tensor) {
315 data_.is_mkl_tensor_ = is_mkl_tensor;
316 }
317
318 inline void SetDimensions(const size_t dimension) {
319 data_.dimension_ = dimension;

Callers 15

ComputeMethod · 0.80
AllocateOutputTensorMethod · 0.80
MklDefaultToEigenMethod · 0.80
AllocateOutputTensorMethod · 0.80
MklDefaultToEigenMethod · 0.80
ComputeMethod · 0.80
AllocateOutputTensorMethod · 0.80
AllocateOutputTensorMethod · 0.80
AllocateOutputTensorMethod · 0.80

Calls

no outgoing calls

Tested by 1

TESTFunction · 0.64