Home
last modified time | relevance | path

Searched refs:mkl_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_dequantize_op_test.cc66 MklDnnShape mkl_shape; in CreateMklInput() local
69 mkl_shape.SetMklTensor(true); in CreateMklInput()
70 mkl_shape.SetMklLayout(&md); in CreateMklInput()
71 mkl_shape.SetElemType(MklDnnType<uint8>()); in CreateMklInput()
72 mkl_shape.SetTfLayout(4, {1, 2, 2, 2}, MKL_TENSOR_FORMAT_NHWC); in CreateMklInput()
76 {static_cast<int64>(mkl_shape.GetSerializeBufferSize())}); in CreateMklInput()
77 mkl_shape.SerializeMklDnnShape( in CreateMklInput()
Dmkl_input_conversion_op.cc192 const MklDnnShape* mkl_shape; in Compute() local
198 mkl_shape = &input_shape_0; in Compute()
204 mkl_shape = &input_shape_1; in Compute()
215 if (mkl_shape->GetTfShape().num_elements() == in Compute()
227 mkl_output_mkl_shape.SetTfLayout(mkl_shape->GetDimension(), in Compute()
228 mkl_shape->GetSizesAsMklDnnDims(), in Compute()
229 mkl_shape->GetTfDataFormat()); in Compute()
231 auto output_mkl_md = mkl_shape->GetMklLayout(); in Compute()
Dmkl_aggregate_ops.cc116 MklDnnShape mkl_shape; in Compute() local
121 GetMklShape(ctx, kSrc0Idx, &mkl_shape); in Compute()
122 bool input_in_mkl_format = mkl_shape.IsMklTensor(); in Compute()
170 GetMklShape(ctx, mkl_input_index, &mkl_shape); in Compute()
172 mkl_data_format = mkl_shape.GetTfDataFormat(); in Compute()
225 output_mkl_shape.SetTfLayout(mkl_shape.GetDimension(), in Compute()
226 mkl_shape.GetSizesAsMklDnnDims(), in Compute()
227 mkl_shape.GetTfDataFormat()); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h742 const MklDnnShape& mkl_shape) { in AllocateOutputSetMklShape() argument
745 second_shape.AddDim(mkl_shape.GetSerializeBufferSize()); in AllocateOutputSetMklShape()
749 mkl_shape.SerializeMklDnnShape( in AllocateOutputSetMklShape()
759 const MklDnnShape& mkl_shape,
767 second_shape.AddDim(mkl_shape.GetSerializeBufferSize());
771 mkl_shape.SerializeMklDnnShape(
901 const MklDnnShape& mkl_shape,
921 AllocateOutputSetMklShape(context, idx_out, mkl_shape);