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

Function GetMklMetaTensor

tensorflow/core/util/mkl_util.h:953–963  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

951// Common utility functions used by OneDNN unit tests
952
953inline Tensor GetMklMetaTensor() {
954 MklDnnShape non_mkl_shape;
955 non_mkl_shape.SetMklTensor(false);
956
957 auto size = static_cast<int64>(non_mkl_shape.GetSerializeBufferSize());
958 Tensor tensor(DT_UINT8, {size});
959
960 non_mkl_shape.SerializeMklDnnShape(tensor.flat<uint8>().data(),
961 size * sizeof(uint8));
962 return tensor;
963}
964
965// -------------------------------------------------------------------
966

Callers 15

SoftmaxFunction · 0.85
CwiseFunction · 0.85
AddNFunction · 0.85
FusedBatchNormFunction · 0.85
FusedBatchNormGradFunction · 0.85
FusedMatMulFunction · 0.85
ConcatFunction · 0.85
IdentityFunction · 0.85
Slice2DFunction · 0.85
LRNFunction · 0.85
LRNGradFunction · 0.85
ActivationFunction · 0.85

Calls 4

SetMklTensorMethod · 0.80
SerializeMklDnnShapeMethod · 0.80
dataMethod · 0.45

Tested by 15

SoftmaxFunction · 0.68
CwiseFunction · 0.68
AddNFunction · 0.68
FusedBatchNormFunction · 0.68
FusedBatchNormGradFunction · 0.68
FusedMatMulFunction · 0.68
ConcatFunction · 0.68
IdentityFunction · 0.68
Slice2DFunction · 0.68
LRNFunction · 0.68
LRNGradFunction · 0.68
ActivationFunction · 0.68