Searched refs:output_tf_format (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_lrn_op.cc | 198 const memory::format& output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor() argument
209 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
448 const memory::format& output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor() argument
459 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
|
| D | mkl_pooling_ops_common.h | 644 const memory::format& output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor() argument
653 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
696 const memory::format& output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor() argument
705 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
|
| D | mkl_conv_grad_input_ops.cc | 537 memory::format output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor() argument
549 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
|
| D | mkl_conv_ops.cc | 1252 memory::format output_tf_format, in AllocateOutputTensor() argument
1269 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
1758 memory::format output_tf_format, in AllocateOutputTensor() argument
1795 output_tf_format, output_tensor); in AllocateOutputTensor()
|
| D | mkl_conv_grad_filter_ops.cc | 668 memory::format output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor() argument
|