Home
last modified time | relevance | path

Searched refs:nhwc (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_requantize_per_channel_op.cc104 memory::format_tag::nhwc); in Compute()
108 memory::format_tag::nhwc) in Compute()
110 memory::format_tag::nhwc); in Compute()
Dmkl_dequantize_op.cc87 memory::format_tag::nhwc); in Compute()
103 memory::format_tag::nhwc); in Compute()
Dmkl_lrn_op.cc138 src_dnn_data.SetOpMemDesc(input_dims, memory::format_tag::nhwc); in Compute()
143 dst_dnn_data.SetOpMemDesc(input_dims, memory::format_tag::nhwc); in Compute()
386 memory::format_tag::nhwc); in Compute()
391 output_dnn_data.SetOpMemDesc(orig_input_dims, memory::format_tag::nhwc); in Compute()
487 memory::format_tag::nhwc); in ConfigureInputGradient()
Dmkl_dequantize_op_test.cc68 memory::desc({1, 2, 2, 2}, MklDnnType<uint8>(), memory::format_tag::nhwc); in CreateMklInput()
Dmkl_quantize_op.cc356 dst_layout_type = memory::format_tag::nhwc; in Compute()
Dmkl_fused_batch_norm_op.cc758 dnn_fmt = memory::format_tag::nhwc; in Compute()
1148 dnn_fmt = memory::format_tag::nhwc; in Compute()
Dmkl_conv_ops.cc1820 memory::format_tag::nhwc); in AllocateOutputTensor()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_conv2d_mkl.cc87 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/kernels/
Dxsmm_conv2d_test.cc51 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()
/external/XNNPACK/
DCMakeLists.txt164 src/operators/argmax-pooling-nhwc.c
165 src/operators/average-pooling-nhwc.c
170 src/operators/convolution-nhwc.c
171 src/operators/deconvolution-nhwc.c
173 src/operators/depth-to-space-nhwc.c
178 src/operators/max-pooling-nhwc.c
181 src/operators/resize-bilinear-nhwc.c
185 src/operators/unpooling-nhwc.c)
3102 ADD_EXECUTABLE(argmax-pooling-nhwc-test test/argmax-pooling-nhwc.cc)
3103 SET_TARGET_PROPERTIES(argmax-pooling-nhwc-test PROPERTIES
[all …]
DAndroid.bp33 "src/operators/argmax-pooling-nhwc.c",
34 "src/operators/average-pooling-nhwc.c",
39 "src/operators/convolution-nhwc.c",
40 "src/operators/deconvolution-nhwc.c",
42 "src/operators/depth-to-space-nhwc.c",
47 "src/operators/max-pooling-nhwc.c",
50 "src/operators/resize-bilinear-nhwc.c",
54 "src/operators/unpooling-nhwc.c",
3533 "test/argmax-pooling-nhwc.cc",
3544 "test/average-pooling-nhwc.cc",
[all …]
DBUILD.bazel54 "src/operators/argmax-pooling-nhwc.c",
55 "src/operators/average-pooling-nhwc.c",
60 "src/operators/convolution-nhwc.c",
61 "src/operators/deconvolution-nhwc.c",
63 "src/operators/depth-to-space-nhwc.c",
68 "src/operators/max-pooling-nhwc.c",
71 "src/operators/resize-bilinear-nhwc.c",
75 "src/operators/unpooling-nhwc.c",
6986 "test/argmax-pooling-nhwc.cc",
6995 "test/average-pooling-nhwc.cc",
[all …]
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1010 if (format == MklTensorFormat::FORMAT_NHWC) return memory::format_tag::nhwc; in MklTensorFormatToMklDnnDataFormat()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc1034 Output nhwc = ops::Reshape( in TEST_F() local
1040 s.WithOpName("flatten"), nhwc, in TEST_F()