Searched refs:input_mkl_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_slice_op.cc | 76 MklDnnShape input_mkl_shape, begin_mkl_shape, size_mkl_shape; in ValidateMklInputs() local
77 GetMklShape(context, kInputTensorIndex, &input_mkl_shape); in ValidateMklInputs()
85 TensorShape input_tf_shape = input_mkl_shape.IsMklTensor() in ValidateMklInputs()
86 ? input_mkl_shape.GetTfShape() in ValidateMklInputs()
148 MklDnnShape input_mkl_shape; in CheckCommonCasesForMklInputs() local
149 GetMklShape(context, 0, &input_mkl_shape); in CheckCommonCasesForMklInputs()
156 AllocateOutputSetMklShape(context, 0, input_mkl_shape); in CheckCommonCasesForMklInputs()
392 MklDnnShape input_mkl_shape; in ComputeMklSlice() local
393 GetMklShape(context, 0, &input_mkl_shape); in ComputeMklSlice()
395 if (input_mkl_shape.IsMklTensor()) { in ComputeMklSlice()
[all …]
|
| D | mkl_pooling_ops_common.h | 605 const MklDnnShape& input_mkl_shape, in ConfigureInput() argument
614 input_mkl_shape.IsMklTensor() in ConfigureInput()
615 ? input_mkl_shape.GetMklLayout() in ConfigureInput()
636 this->InitMklPoolParameters(context, pool_params, input_mkl_shape, in ConfigureInput()
664 const MklDnnShape& input_mkl_shape) { in SanityCheckInput() argument
665 if (!input_mkl_shape.IsMklTensor()) { in SanityCheckInput()
671 input_mkl_shape.GetDimension() == 4 || in SanityCheckInput()
672 input_mkl_shape.GetDimension() == 5, in SanityCheckInput()
|
| D | mkl_conv_grad_input_ops.cc | 484 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes() argument
490 CHECK(!input_mkl_shape.IsMklTensor()) in ValidateMklShapes()
|
| D | mkl_conv_grad_filter_ops.cc | 612 void ValidateMklShapes(const MklDnnShape& input_mkl_shape, in ValidateMklShapes() argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 949 MklDnnShape input_mkl_shape; in GetTfShape() local
950 GetMklShape(context, input_idx, &input_mkl_shape); in GetTfShape()
951 if (input_mkl_shape.IsMklTensor()) { in GetTfShape()
952 return input_mkl_shape.GetTfShape(); in GetTfShape()
|