Home
last modified time | relevance | path

Searched refs:FORMAT_NHWC (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc48 if (data_format == FORMAT_NHWC) { in Compute()
61 if (data_format == FORMAT_NHWC) { in Compute()
108 if (data_format == FORMAT_NHWC) { in Compute()
135 if (data_format == FORMAT_NHWC || tensor_in == nullptr) { in Compute()
146 if (data_format == FORMAT_NHWC || tensor_out == nullptr) { in Compute()
156 if (data_format == FORMAT_NHWC) { in Compute()
165 if (data_format == FORMAT_NHWC) { in Compute()
173 if (data_format == FORMAT_NHWC) { in Compute()
237 if (data_format == FORMAT_NHWC) { in Compute()
Dpooling_ops_common.cc180 if (data_format == FORMAT_NHWC) { in Compute()
193 if (data_format == FORMAT_NHWC) { in Compute()
208 case FORMAT_NHWC: in Compute()
272 if (data_format == FORMAT_NHWC) { in Compute()
313 if (data_format == FORMAT_NHWC || !tensor_in) { in Compute()
324 if (data_format == FORMAT_NHWC || !tensor_out) { in Compute()
334 if (data_format == FORMAT_NHWC) { in Compute()
343 if (data_format == FORMAT_NHWC) { in Compute()
352 if (data_format == FORMAT_NHWC) { in Compute()
425 if (data_format == FORMAT_NHWC) { in Compute()
Dspacetodepth_op_gpu.cu.cc144 struct SpaceToDepthOpFunctor<GPUDevice, T, FORMAT_NHWC> {
240 template struct functor::SpaceToDepthOpFunctor<GPUDevice, float, FORMAT_NHWC>;
246 FORMAT_NHWC>;
250 template struct functor::SpaceToDepthOpFunctor<GPUDevice, uint8, FORMAT_NHWC>;
Ddepthtospace_op.cc59 context, data_format_ == FORMAT_NHWC, in DepthToSpaceOp()
136 functor::DepthToSpaceOpFunctor<Device, T, FORMAT_NHWC> functor; in Compute()
150 struct DepthToSpaceOpFunctor<CPUDevice, T, FORMAT_NHWC> {
Dspacetodepth_op.cc59 context, data_format_ == FORMAT_NHWC, in SpaceToDepthOp()
138 functor::SpaceToDepthOpFunctor<Device, T, FORMAT_NHWC> functor; in Compute()
151 struct SpaceToDepthOpFunctor<CPUDevice, T, FORMAT_NHWC> {
Dbias_op_gpu.cu.cc87 if (data_format == FORMAT_NHWC) { in compute()
212 if (data_format == FORMAT_NHWC) { in compute()
217 if (data_format == FORMAT_NHWC) { in compute()
238 if (data_format == FORMAT_NHWC) { in compute()
Dconv_grad_ops_3d.cc169 data_format_(FORMAT_NHWC), in Conv3DBackpropInputOp()
178 context, data_format_ == FORMAT_NHWC, in Conv3DBackpropInputOp()
275 data_format_(FORMAT_NHWC), in Conv3DCustomBackpropInputOp()
284 context, data_format_ == FORMAT_NHWC, in Conv3DCustomBackpropInputOp()
625 data_format_(FORMAT_NHWC), in Conv3DBackpropFilterOp()
634 context, data_format_ == FORMAT_NHWC, in Conv3DBackpropFilterOp()
737 data_format_(FORMAT_NHWC), in Conv3DCustomBackpropFilterOp()
746 context, data_format_ == FORMAT_NHWC, in Conv3DCustomBackpropFilterOp()
1094 data_format_(FORMAT_NHWC), in Conv3DBackpropInputOp()
1171 data_format_ == FORMAT_NHWC) { in Compute()
[all …]
Ddepthtospace_op_gpu.cu.cc148 struct DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NHWC> {
244 template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NHWC>;
250 FORMAT_NHWC>;
Dmaxpooling_op.cc220 context, data_format_ == FORMAT_NHWC, in MaxPoolingGradOp()
299 padding_, FORMAT_NHWC, tensor_in.shape()}; in Compute()
332 padding, FORMAT_NHWC, tensor_in_shape}; in MaxPoolingBackwardCustomKernel()
425 CHECK(data_format_ == FORMAT_NHWC) in Compute()
459 context, data_format_ == FORMAT_NHWC, in MaxPoolingGradGradOp()
528 padding_, FORMAT_NHWC, tensor_in.shape()}; in Compute()
762 context, data_format_ == FORMAT_NHWC, in MaxPoolingNoMaskOp()
815 context, data_format_ == FORMAT_NHWC, in MaxPoolingNoMaskV2Op()
925 padding_, FORMAT_NHWC, tensor_in.shape()}; in Compute()
967 GetTensorDim(grad_out->shape(), FORMAT_NHWC, 'N'); in launch()
[all …]
Dconv_ops_3d.cc57 OP_REQUIRES(context, data_format == FORMAT_NHWC, in launch()
242 data_format == FORMAT_NHWC) { in launch()
268 data_format == FORMAT_NHWC) { in launch()
328 if (data_format == FORMAT_NHWC) { in launch()
499 if (data_format == FORMAT_NHWC) { in launch()
Dconv_ops.cc74 CHECK(data_format == FORMAT_NHWC) << "Generic conv implementation only " in operator ()()
130 if (data_format != FORMAT_NHWC) { in operator ()()
177 if (data_format != FORMAT_NHWC || dilation_rows != 1 || in Run()
612 col_stride == 1 && data_format == FORMAT_NHWC && in operator ()()
640 data_format == FORMAT_NHWC) { in operator ()()
741 if (data_format == FORMAT_NHWC) { in operator ()()
801 if (data_format == FORMAT_NHWC) { in operator ()()
910 if (data_format == FORMAT_NHWC) { in operator ()()
Dfused_batch_norm_op.cc75 OP_REQUIRES(context, tensor_format == FORMAT_NHWC, in operator ()()
161 OP_REQUIRES(context, tensor_format == FORMAT_NHWC, in operator ()()
279 } else if (tensor_format == FORMAT_NHWC) { in operator ()()
374 if (tensor_format == FORMAT_NHWC) { in operator ()()
418 } else if (tensor_format == FORMAT_NHWC) { in operator ()()
497 if (tensor_format == FORMAT_NHWC) { in operator ()()
685 CHECK(tensor_format_ == FORMAT_NHWC) in Compute()
Dpooling_ops_common.h88 context, data_format_ == FORMAT_NHWC, in MaxPoolingOp()
93 data_format_ = FORMAT_NHWC; in MaxPoolingOp()
112 padding_, FORMAT_NHWC, tensor_in.shape()}; in Compute()
299 data_format_ == FORMAT_NHWC || data_format_ == FORMAT_NCHW_VECT_C,
304 data_format_ = FORMAT_NHWC;
Dconv_ops_fused_image_transform.cc636 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N'); in FusedResizeConv2DUsingGemmOp()
637 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C'); in FusedResizeConv2DUsingGemmOp()
822 const int stride_rows = GetTensorDim(strides_, FORMAT_NHWC, 'H'); in Compute()
823 const int stride_cols = GetTensorDim(strides_, FORMAT_NHWC, 'W'); in Compute()
833 ShapeFromFormat(FORMAT_NHWC, batch, out_rows, out_cols, out_depth); in Compute()
Dbias_op.cc54 if (data_format == FORMAT_NHWC) { in GetBiasValueDims()
96 data_format_ = FORMAT_NHWC; in BiasOp()
241 data_format_ = FORMAT_NHWC; in BiasGradOp()
339 data_format_ = FORMAT_NHWC; in BiasOp()
Dconv_ops_fused_impl.h321 OP_REQUIRES(context, params.data_format == FORMAT_NHWC,
563 params.data_format == FORMAT_NHWC || params.data_format == FORMAT_NCHW,
646 if (params.data_format == FORMAT_NHWC) {
723 if (params.data_format == FORMAT_NHWC) {
816 if (params.data_format == FORMAT_NHWC) {
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc37 data_format_ = FORMAT_NHWC; in BiasOp()
53 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile()
86 data_format_ = FORMAT_NHWC; in BiasAddGradOp()
98 int feature_dim = (data_format_ == FORMAT_NHWC) in Compile()
Ddepthtospace_op.cc53 OP_REQUIRES(ctx, data_format == FORMAT_NCHW || data_format == FORMAT_NHWC, in Compile()
78 if (data_format == FORMAT_NHWC) { in Compile()
Dspacetodepth_op.cc53 OP_REQUIRES(ctx, data_format == FORMAT_NCHW || data_format == FORMAT_NHWC, in Compile()
78 if (data_format == FORMAT_NHWC) { in Compile()
Dpooling_ops.cc128 TensorFormat data_format_ = FORMAT_NHWC;
349 TensorFormat data_format_ = FORMAT_NHWC;
451 TensorFormat data_format_ = FORMAT_NHWC;
622 TensorFormat data_format_ = FORMAT_NHWC;
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h37 FORMAT_NHWC = 0, enumerator
113 case FORMAT_NHWC: in GetTensorSpatialDims()
139 case FORMAT_NHWC: in GetTensorDimsFromSpatialDims()
164 case FORMAT_NHWC: in GetTensorBatchDimIndex()
184 case FORMAT_NHWC: in GetTensorFeatureDimIndex()
221 case FORMAT_NHWC: in GetTensorSpatialDimIndex()
303 if (format == FORMAT_NHWC || format == FORMAT_NHWC_VECT_W) { in GetTensorDimIndex()
Dtensor_format.cc42 case FORMAT_NHWC: in ToString()
76 *format = FORMAT_NHWC; in FormatFromString()
Dtensor_format_test.cc29 EnumStringPair(FORMAT_NHWC), EnumStringPair(FORMAT_NCHW),
117 (format == FORMAT_NHWC || in GetTensorDimMap()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc679 case FORMAT_NHWC: in ImageDims()
1149 ImageDims(FORMAT_NHWC, batch, feature, input_dims)) in TEST_F()
1151 .Attr("ksize", ImageDims(FORMAT_NHWC, 1, 1, kernel_dims)) in TEST_F()
1152 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, stride_dims)) in TEST_F()
1164 AsInt32s(ImageDims(FORMAT_NHWC, batch, features, d.input_dims)); in TEST_F()
1166 ImageDims(FORMAT_NHWC, batch, features, d.output_dims); in TEST_F()
1172 .Attr("ksize", ImageDims(FORMAT_NHWC, 1, 1, d.kernel_dims)) in TEST_F()
1173 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, d.stride_dims)) in TEST_F()
1184 AsInt32s(ImageDims(FORMAT_NHWC, batch, features, d.input_dims)); in TEST_F()
1186 ImageDims(FORMAT_NHWC, batch, features, d.output_dims); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dmkl_tfconversion_pass.cc182 (data_format == ToString(FORMAT_NHWC) || in InsertConversionNodeOnEdge()
267 (data_format == ToString(FORMAT_NHWC) || in InsertInputConversionNode()

12