Searched defs:output_dims_mkl_order (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_avgpooling_op.cc | 73 memory::dims output_dims_mkl_order; in Compute() local
223 memory::dims output_dims_mkl_order; in Compute() local
|
| D | mkl_maxpooling_op.cc | 81 memory::dims output_dims_mkl_order; in Compute() local
280 memory::dims output_dims_mkl_order; in Compute() local
|
| D | mkl_pooling_ops_common.h | 481 memory::dims* output_dims_mkl_order) { in GetOutputDims()
546 const memory::dims output_dims_mkl_order, in AllocateEmptyOutputTensor()
644 const memory::dims output_dims_mkl_order, in AllocateOutputTensor()
698 const memory::dims output_dims_mkl_order, in AllocateOutputTensor()
|
| D | mkl_lrn_op.cc | 192 const memory::dims output_dims_mkl_order, in AllocateOutputTensor()
455 const memory::dims output_dims_mkl_order, in AllocateOutputTensor()
|
| D | mkl_conv_ops.h | 488 memory::dims* output_dims_mkl_order, memory::dims* pad_l, in GetOutputAndPadSizeInMklOrder()
|
| D | mkl_conv_grad_input_ops.cc | 561 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor()
|
| D | mkl_conv_grad_filter_ops.cc | 675 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor()
|
| D | mkl_matmul_ops_common.h | 392 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor()
|
| D | mkl_conv_ops.cc | 850 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor()
1740 const memory::dims& output_dims_mkl_order, in AllocateOutputTensor()
|