Searched refs:TFShapeToMklDnnDimsInNCHW (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_aggregate_ops.cc | 143 src2_dims = TFShapeToMklDnnDimsInNCHW(src2_tensor.shape(), in Compute()
157 src1_dims = TFShapeToMklDnnDimsInNCHW(src1_tensor.shape(), in Compute()
|
| D | mkl_maxpooling_op.cc | 552 TFShapeToMklDnnDimsInNCHW( in Compute()
564 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(input_tensor.shape(), in Compute()
717 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(orig_input_shape, in Compute()
725 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(grad_tensor.shape(), in Compute()
|
| D | mkl_avgpooling_op.cc | 486 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(input_tensor.shape(), in Compute()
606 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(orig_input_shape, in Compute()
614 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(grad_tensor.shape(), in Compute()
|
| D | mkl_pooling_ops_common.h | 618 ? TFShapeToMklDnnDimsInNCHW(input_tensor_shape, in ConfigureInput()
725 ? TFShapeToMklDnnDimsInNCHW(input_gradient_tensor.shape(), in ConfigureInputGradient()
745 : TFShapeToMklDnnDimsInNCHW(input_gradient_tensor.shape(), in ConfigureInputGradient()
|
| D | mkl_requantize_per_channel_op.cc | 97 TFShapeToMklDnnDimsInNCHW(input.shape(), FORMAT_NHWC); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 613 : TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), tensor_format_); in Compute()
927 : TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), tensor_format_); in Compute()
931 : TFShapeToMklDnnDimsInNCHW(diff_dst_tensor.shape(), in Compute()
|
| D | mkl_relu_op.cc | 602 auto diff_dst_dims = TFShapeToMklDnnDimsInNCHW(diff_dst_tensor.shape(), in Compute()
616 ? TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1516 inline memory::dims TFShapeToMklDnnDimsInNCHW(const TensorShape& shape, in TFShapeToMklDnnDimsInNCHW() function
|