Searched refs:output_tf_format (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 193 const MklTensorFormat& output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor() argument
204 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
456 const MklTensorFormat& output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor() argument
466 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
|
| D | mkl_pooling_ops_common.h | 645 const MklTensorFormat& output_tf_format, in AllocateOutputTensor() argument
656 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
699 const MklTensorFormat& output_tf_format, in AllocateOutputTensor() argument
708 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
|
| D | mkl_conv_grad_input_ops.cc | 562 MklTensorFormat output_tf_format, in AllocateOutputTensor() argument
575 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
|
| D | mkl_matmul_ops_common.h | 393 MklTensorFormat output_tf_format, Tensor** output_tensor) { in AllocateOutputTensor() argument
402 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
|
| D | mkl_conv_ops.cc | 851 MklTensorFormat output_tf_format, in AllocateOutputTensor() argument
867 output_dims_mkl_order, output_tf_format); in AllocateOutputTensor()
1741 MklTensorFormat output_tf_format, in AllocateOutputTensor() argument
1778 output_tf_format, output_mkl_shape, in AllocateOutputTensor()
|