Searched refs:nchw (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| 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
53 LIBXSMM_VLA_DECL(4, const float, input, nchw, C, H, W); in naive_copy_NCHW_to_NHWC()
|
| D | mkl_softmax_op.cc | 97 layout_type = memory::format::nchw; in Compute()
|
| D | mkl_concat_op.cc | 339 memory::desc(src_dims, MklDnnType<T>(), memory::format::nchw); in Compute()
364 dst_md = memory::desc(dst_dims, MklDnnType<T>(), memory::format::nchw); in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 601 format_m = memory::format::nchw; in Compute()
912 format_m = memory::format::nchw; in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 36 a_mkldnn_shape.SetTfLayout(a_dims.size(), a_dims, memory::format::nchw); in TEST()
|
| D | mkl_util.h | 1468 return memory::format::nchw; in TFDataFormatToMklDnnDataFormat()
1481 else if (format == memory::format::nchw || format == memory::format::ncdhw) in MklDnnDataFormatToTFDataFormat()
2253 fmt_desired = is_2d ? memory::format::nchw : memory::format::ncdhw;
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers_test.py | 2718 nchw = self._runBatchNormalizationWithFormat(
2720 self.assertAllClose(nhwc, nchw, atol=1e-4, rtol=1e-4)
2727 nchw = self._runBatchNormalizationWithFormat(
2729 self.assertAllClose(nhwc, nchw, atol=1e-4, rtol=1e-4)
|