Searched refs:TFShapeToMklDnnDimsInNCDHW (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_maxpooling_op.cc | 114 TFShapeToMklDnnDimsInNCDHW( in Compute()
124 : TFShapeToMklDnnDimsInNCDHW(input_tensor.shape(), in Compute()
269 : TFShapeToMklDnnDimsInNCDHW(orig_input_shape, in Compute()
277 : TFShapeToMklDnnDimsInNCDHW(grad_tensor.shape(), in Compute()
|
| D | mkl_avgpooling_op.cc | 95 : TFShapeToMklDnnDimsInNCDHW(input_tensor.shape(), in Compute()
213 : TFShapeToMklDnnDimsInNCDHW(orig_input_shape, in Compute()
221 : TFShapeToMklDnnDimsInNCDHW(grad_tensor.shape(), in Compute()
|
| D | mkl_aggregate_ops.cc | 200 src_dims = TFShapeToMklDnnDimsInNCDHW(src_tensor.shape(), in Compute()
|
| D | mkl_pooling_ops_common.h | 622 : TFShapeToMklDnnDimsInNCDHW(input_tensor_shape, in ConfigureInput()
|
| D | mkl_relu_op.cc | 655 : TFShapeToMklDnnDimsInNCDHW(src_tensor.shape(), in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1102 inline memory::dims TFShapeToMklDnnDimsInNCDHW(const TensorShape& shape, in TFShapeToMklDnnDimsInNCDHW() function
|