Searched refs:nhwc (Results 1 - 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_util_test.cc | 48 b_mkldnn_shape.SetTfLayout(b_dims.size(), b_dims, memory::format::nhwc);
|
| H A D | mkl_util.h | 1268 return memory::format::nhwc;
1282 if (format == memory::format::nhwc)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | xsmm_conv2d_test.cc | 51 LIBXSMM_INLINE void naive_copy_NCHW_to_NHWC(const float* nchw, Tensor& nhwc, argument
55 auto output = nhwc.flat<float>();
|
| H A D | mkl_lrn_op.cc | 804 src_dnn_data.SetOpMemDesc(input_dims, memory::format::nhwc);
808 dst_dnn_data.SetOpMemDesc(input_dims, memory::format::nhwc);
1060 orig_input_dnn_data.SetOpMemDesc(orig_input_dims, memory::format::nhwc);
1064 output_dnn_data.SetOpMemDesc(orig_input_dims, memory::format::nhwc);
1150 input_grad_dnn_data->SetOpMemDesc(orig_input_dims, memory::format::nhwc);
|
| H A D | mkl_fused_batch_norm_op.cc | 797 format_m = memory::format::nhwc;
1123 format_m = memory::format::nhwc;
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | arithmetic_optimizer_test.cc | 694 Output nhwc = ops::Reshape( local
695 s.WithOpName("nhwc"), transpose,
698 s.WithOpName("flatten"), nhwc,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| H A D | layers_test.py | 2680 nhwc = self._runBatchNormalizationWithFormat(
2684 self.assertAllClose(nhwc, nchw, atol=1e-4, rtol=1e-4)
2689 nhwc = self._runBatchNormalizationWithFormat(
2693 self.assertAllClose(nhwc, nchw, atol=1e-4, rtol=1e-4)
|
Completed in 918 milliseconds