Searched refs:TFShapeToMklDnnDimsInNCHW (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_maxpooling_op.cc | 110 TFShapeToMklDnnDimsInNCHW( in Compute()
122 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(input_tensor.shape(), in Compute()
267 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(orig_input_shape, in Compute()
275 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(grad_tensor.shape(), in Compute()
|
| D | mkl_avgpooling_op.cc | 93 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(input_tensor.shape(), in Compute()
211 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(orig_input_shape, in Compute()
219 : is_pool2d ? TFShapeToMklDnnDimsInNCHW(grad_tensor.shape(), in Compute()
|
| D | mkl_requantize_per_channel_op.cc | 102 TFShapeToMklDnnDimsInNCHW(input.shape(), FORMAT_NHWC); in Compute()
|
| D | mkl_aggregate_ops.cc | 197 TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), tf_data_format); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 770 : TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), tensor_format_); in Compute()
1164 : TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), tensor_format_); in Compute()
1168 : TFShapeToMklDnnDimsInNCHW(diff_dst_tensor.shape(), in Compute()
|
| D | mkl_relu_op.cc | 638 auto diff_dst_dims = TFShapeToMklDnnDimsInNCHW(diff_dst_tensor.shape(), in Compute()
653 ? TFShapeToMklDnnDimsInNCHW(src_tensor.shape(), in Compute()
|
| D | mkl_pooling_ops_common.h | 620 ? TFShapeToMklDnnDimsInNCHW(input_tensor_shape, in ConfigureInput()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1087 inline memory::dims TFShapeToMklDnnDimsInNCHW(const TensorShape& shape, in TFShapeToMklDnnDimsInNCHW() function
|