Searched refs:TFShapeToMklDnnDimsInNCDHW (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_aggregate_ops.cc | 146 src2_dims = TFShapeToMklDnnDimsInNCDHW(src2_tensor.shape(), in Compute()
160 src1_dims = TFShapeToMklDnnDimsInNCDHW(src1_tensor.shape(), in Compute()
|
| D | mkl_maxpooling_op.cc | 556 TFShapeToMklDnnDimsInNCDHW( in Compute()
566 : TFShapeToMklDnnDimsInNCDHW(input_tensor.shape(), in Compute()
719 : TFShapeToMklDnnDimsInNCDHW(orig_input_shape, in Compute()
727 : TFShapeToMklDnnDimsInNCDHW(grad_tensor.shape(), in Compute()
|
| D | mkl_avgpooling_op.cc | 488 : TFShapeToMklDnnDimsInNCDHW(input_tensor.shape(), in Compute()
608 : TFShapeToMklDnnDimsInNCDHW(orig_input_shape, in Compute()
616 : TFShapeToMklDnnDimsInNCDHW(grad_tensor.shape(), in Compute()
|
| D | mkl_pooling_ops_common.h | 620 : TFShapeToMklDnnDimsInNCDHW(input_tensor_shape, in ConfigureInput()
727 : TFShapeToMklDnnDimsInNCDHW( in ConfigureInputGradient()
|
| D | mkl_relu_op.cc | 618 : TFShapeToMklDnnDimsInNCDHW(src_tensor.shape(), in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1531 inline memory::dims TFShapeToMklDnnDimsInNCDHW(const TensorShape& shape, in TFShapeToMklDnnDimsInNCDHW() function
|