Searched refs:nhwc (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_requantize_per_channel_op.cc | 99 memory::format::nhwc); in Compute()
103 memory::format::nhwc) in Compute()
105 memory::format::nhwc); in Compute()
|
| D | mkl_lrn_op.cc | 134 src_dnn_data.SetOpMemDesc(input_dims, memory::format::nhwc); in Compute()
138 dst_dnn_data.SetOpMemDesc(input_dims, memory::format::nhwc); in Compute()
389 orig_input_dnn_data.SetOpMemDesc(orig_input_dims, memory::format::nhwc); in Compute()
393 output_dnn_data.SetOpMemDesc(orig_input_dims, memory::format::nhwc); in Compute()
479 input_grad_dnn_data->SetOpMemDesc(orig_input_dims, memory::format::nhwc); in ConfigureInputGradient()
|
| D | xsmm_conv2d_test.cc | 51 LIBXSMM_INLINE void naive_copy_NCHW_to_NHWC(const float* nchw, Tensor& nhwc, in naive_copy_NCHW_to_NHWC() argument
55 auto output = nhwc.flat<float>(); in naive_copy_NCHW_to_NHWC()
|
| D | mkl_softmax_op.cc | 95 layout_type = memory::format::nhwc; in Compute()
|
| D | mkl_pooling_ops_common.cc | 53 context_.src_fmt = is_2d ? memory::format::nhwc : memory::format::ndhwc; in Setup()
|
| D | mkl_fused_batch_norm_op.cc | 603 format_m = memory::format::nhwc; in Compute()
914 format_m = memory::format::nhwc; in Compute()
|
| D | mkl_conv_ops.cc | 1825 memory::format::nhwc); in AllocateOutputTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_mkl.cc | 87 memory({{{conv1_src_dim}, memory::data_type::f32, memory::format::nhwc}, in MKLConvImpl()
95 memory({{{conv1_dst_dim}, memory::data_type::f32, memory::format::nhwc}, in MKLConvImpl()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 48 b_mkldnn_shape.SetTfLayout(b_dims.size(), b_dims, memory::format::nhwc); in TEST()
|
| D | mkl_util.h | 1466 return memory::format::nhwc; in TFDataFormatToMklDnnDataFormat()
1479 if (format == memory::format::nhwc || format == memory::format::ndhwc) in MklDnnDataFormatToTFDataFormat()
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers_test.py | 2716 nhwc = self._runBatchNormalizationWithFormat(
2720 self.assertAllClose(nhwc, nchw, atol=1e-4, rtol=1e-4)
2725 nhwc = self._runBatchNormalizationWithFormat(
2729 self.assertAllClose(nhwc, nchw, atol=1e-4, rtol=1e-4)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 1087 Output nhwc = ops::Reshape( in TEST_F() local
1091 s.WithOpName("flatten"), nhwc, in TEST_F()
|