Searched defs:output_dims_mkl_order (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_pooling_ops_common.h | 491 memory::dims* output_dims_mkl_order) { in GetOutputDims()
556 const memory::dims output_dims_mkl_order, in AllocateEmptyOutputTensor()
643 const memory::dims output_dims_mkl_order, in AllocateOutputTensor()
695 const memory::dims output_dims_mkl_order, in AllocateOutputTensor()
|
| D | mkl_avgpooling_op.cc | 466 memory::dims output_dims_mkl_order; in Compute() local
618 memory::dims output_dims_mkl_order; in Compute() local
|
| D | mkl_lrn_op.cc | 197 const memory::dims output_dims_mkl_order, in AllocateOutputTensor()
447 const memory::dims output_dims_mkl_order, in AllocateOutputTensor()
|
| D | mkl_maxpooling_op.cc | 536 memory::dims output_dims_mkl_order; in Compute() local
730 memory::dims output_dims_mkl_order; in Compute() local
|
| D | mkl_conv_ops.h | 479 memory::dims* output_dims_mkl_order, memory::dims* pad_l, in GetOutputAndPadSizeInMklOrder()
|
| D | mkl_conv_grad_input_ops.cc | 536 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor()
|
| D | mkl_conv_grad_filter_ops.cc | 667 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor()
|
| D | mkl_conv_ops.cc | 1251 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor()
1757 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor()
|