Searched refs:input_mkl_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 73 MklDnnShape input_mkl_shape, begin_mkl_shape, size_mkl_shape; in ValidateMklInputs() local
74 GetMklShape(context, kInputTensorIndex, &input_mkl_shape); in ValidateMklInputs()
82 TensorShape input_tf_shape = input_mkl_shape.IsMklTensor() in ValidateMklInputs()
83 ? input_mkl_shape.GetTfShape() in ValidateMklInputs()
146 MklDnnShape input_mkl_shape; in CheckCommonCasesForMklInputs() local
147 GetMklShape(context, 0, &input_mkl_shape); in CheckCommonCasesForMklInputs()
154 AllocateOutputSetMklShape(context, 0, input_mkl_shape); in CheckCommonCasesForMklInputs()
401 MklDnnShape input_mkl_shape; in ComputeMklSlice() local
402 GetMklShape(context, 0, &input_mkl_shape); in ComputeMklSlice()
404 if (input_mkl_shape.IsMklTensor()) { in ComputeMklSlice()
[all …]
|
| D | mkl_pooling_ops_common.h | 607 const MklDnnShape& input_mkl_shape, in ConfigureInput() argument
616 input_mkl_shape.IsMklTensor() in ConfigureInput()
617 ? input_mkl_shape.GetMklLayout() in ConfigureInput()
638 this->InitMklPoolParameters(context, pool_params, input_mkl_shape, in ConfigureInput()
670 const MklDnnShape& input_mkl_shape) { in SanityCheckInput() argument
671 if (!input_mkl_shape.IsMklTensor()) { in SanityCheckInput()
677 input_mkl_shape.GetDimension() == 4 || in SanityCheckInput()
678 input_mkl_shape.GetDimension() == 5, in SanityCheckInput()
|
| D | mkl_conv_grad_input_ops.cc | 511 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes() argument
517 CHECK(!input_mkl_shape.IsMklTensor()) in ValidateMklShapes()
|
| D | mkl_conv_grad_filter_ops.cc | 623 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes() argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_types.h | 115 #define SKIP_INPUT_REORDER(input_mkl_shape, input_md) \ argument
116 input_mkl_shape.GetTfDataFormat() == MKL_TENSOR_FORMAT_BLOCKED
|
| D | mkl_util.h | 625 const MklDnnShape& input_mkl_shape, in ConvertMklToTF() argument
628 if (!input_mkl_shape.IsMklTensor()) { in ConvertMklToTF()
635 TensorShape output_tf_shape = input_mkl_shape.GetTfShape(); in ConvertMklToTF()
643 auto input_mkl_md = input_mkl_shape.GetMklLayout(); in ConvertMklToTF()
644 auto output_tf_md = input_mkl_shape.GetTfLayout(); in ConvertMklToTF()
729 MklDnnShape input_mkl_shape; variable
730 GetMklShape(context, input_idx, &input_mkl_shape, eager_mode);
731 if (input_mkl_shape.IsMklTensor() && !eager_mode) {
732 return input_mkl_shape.GetTfShape();
|