Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc48 if (data_format == FORMAT_NHWC) { in Compute()
61 if (data_format == FORMAT_NHWC) { in Compute()
123 if (data_format == FORMAT_NHWC) { in Compute()
150 if (data_format == FORMAT_NHWC || tensor_in == nullptr) { in Compute()
161 if (data_format == FORMAT_NHWC || tensor_out == nullptr) { in Compute()
171 if (data_format == FORMAT_NHWC) { in Compute()
180 if (data_format == FORMAT_NHWC) { in Compute()
188 if (data_format == FORMAT_NHWC) { in Compute()
268 if (data_format == FORMAT_NHWC) { in Compute()
Dfused_batch_norm_op.cu.cc131 static_assert(tensor_format == FORMAT_NHWC || tensor_format == FORMAT_NCHW,
146 const int channel = (tensor_format == FORMAT_NHWC) in run()
222 if (tensor_format == FORMAT_NHWC) { in run()
262 const int32 channel = (tensor_format == FORMAT_NHWC) in run()
365 if (tensor_format == FORMAT_NHWC) { in operator ()()
369 LAUNCH(FORMAT_NHWC, false, Activation::kIdentity, c, 1); in operator ()()
371 LAUNCH(FORMAT_NHWC, false, Activation::kRelu, c, 1); in operator ()()
373 LAUNCH(FORMAT_NHWC, true, Activation::kIdentity, c, 1); in operator ()()
375 LAUNCH(FORMAT_NHWC, true, Activation::kRelu, c, 1); in operator ()()
Dspacetodepth_op.cc74 context, data_format_ == FORMAT_NHWC, in SpaceToDepthOp()
145 functor::SpaceToDepthOpFunctor<Device, RT, FORMAT_NHWC> functor; in Compute()
152 functor::SpaceToDepthOpFunctor<Device, T, FORMAT_NHWC> functor; in Compute()
166 struct SpaceToDepthOpFunctor<CPUDevice, T, FORMAT_NHWC> {
Dpooling_ops_common.cc244 if (data_format == FORMAT_NHWC) { in Compute()
257 if (data_format == FORMAT_NHWC) { in Compute()
272 case FORMAT_NHWC: in Compute()
410 if (data_format == FORMAT_NHWC) { in Compute()
473 if (data_format == FORMAT_NHWC || !tensor_in) { in Compute()
484 if (data_format == FORMAT_NHWC || !tensor_out) { in Compute()
494 if (data_format == FORMAT_NHWC) { in Compute()
503 if (data_format == FORMAT_NHWC) { in Compute()
512 if (data_format == FORMAT_NHWC) { in Compute()
557 case FORMAT_NHWC: in Compute()
[all …]
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>;
Dconv_ops_benchmark_test.cc78 TensorFormat data_format = FORMAT_NHWC) { in Conv2D()
81 Tensor images_t = data_format == FORMAT_NHWC in Conv2D()
109 int out_depth, TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBias()
136 TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBiasAndActivation()
158 int out_depth, TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBatchNorm()
196 TensorFormat data_format = FORMAT_NHWC) { in Conv2DWithBatchNormAndActivation()
221 TensorFormat data_format = FORMAT_NHWC) {
224 Tensor images_t = data_format == FORMAT_NHWC
260 TensorFormat data_format = FORMAT_NHWC) {
263 Tensor images_t = data_format == FORMAT_NHWC
Dconv_grad_ops_3d.cc178 data_format_(FORMAT_NHWC), in Conv3DBackpropInputOp()
187 context, data_format_ == FORMAT_NHWC, in Conv3DBackpropInputOp()
285 data_format_(FORMAT_NHWC), in Conv3DCustomBackpropInputOp()
294 context, data_format_ == FORMAT_NHWC, in Conv3DCustomBackpropInputOp()
636 data_format_(FORMAT_NHWC), in Conv3DBackpropFilterOp()
645 context, data_format_ == FORMAT_NHWC, in Conv3DBackpropFilterOp()
748 data_format_(FORMAT_NHWC), in Conv3DCustomBackpropFilterOp()
757 context, data_format_ == FORMAT_NHWC, in Conv3DCustomBackpropFilterOp()
1107 data_format_(FORMAT_NHWC), in Conv3DBackpropInputOp()
1185 dims.stride(2) == 1 && data_format_ == FORMAT_NHWC) { in Compute()
[all …]
Ddepthtospace_op.cc59 context, data_format_ == FORMAT_NHWC, in DepthToSpaceOp()
135 functor::DepthToSpaceOpFunctor<Device, T, FORMAT_NHWC> functor; in Compute()
149 struct DepthToSpaceOpFunctor<CPUDevice, T, FORMAT_NHWC> {
Dfused_batch_norm_ex_op_test.cc344 data_format == FORMAT_NHWC in VerifyTensorsNear()
476 this->VerifyFusedBatchNormEx(4, 28, 28, 256, FORMAT_NHWC, kInTraining, in TYPED_TEST_P()
481 this->VerifyFusedBatchNormEx(4, 28, 28, 256, FORMAT_NHWC, kInTraining, in TYPED_TEST_P()
487 this->VerifyFusedBatchNormEx(4, 28, 28, 256, FORMAT_NHWC, kInTraining, in TYPED_TEST_P()
512 this->VerifyFusedBatchNormEx(4, 28, 28, 256, FORMAT_NHWC, kInInference, in TYPED_TEST_P()
517 this->VerifyFusedBatchNormEx(4, 28, 28, 256, FORMAT_NHWC, kInInference, in TYPED_TEST_P()
523 this->VerifyFusedBatchNormEx(4, 28, 28, 256, FORMAT_NHWC, kInInference, in TYPED_TEST_P()
556 Tensor x_t(dtype, data_format == FORMAT_NHWC ? TensorShape({n, h, w, c}) in FusedBatchNormEx()
Dfused_batch_norm_op.cc128 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
133 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
266 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
271 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
358 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
363 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
368 ShapeFromFormat(FORMAT_NHWC, in_batch, in operator ()()
797 fast_nhwc_batch_norm && tensor_format == FORMAT_NHWC ? FORMAT_NHWC in operator ()()
861 } else if (tensor_format == FORMAT_NHWC && compute_format == FORMAT_NCHW) { in operator ()()
887 compute_format == FORMAT_NHWC ? se::dnn::DataLayout::kBatchYXDepth in operator ()()
[all …]
Dconv_grad_input_ops.cc134 data_format == FORMAT_NHWC && (padding == VALID || padding == SAME)) { in operator ()()
166 data_format == FORMAT_NHWC) { in operator ()()
229 (compute_in_nhwc && data_format == FORMAT_NHWC) ? FORMAT_NHWC in operator ()()
247 compute_data_format == FORMAT_NHWC ? kComputeInNHWC : kComputeInNCHW; in operator ()()
305 } else if (compute_data_format == FORMAT_NHWC) { in operator ()()
314 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
540 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
Ddepthtospace_op_gpu.cu.cc147 struct DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NHWC> {
243 template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NHWC>;
249 FORMAT_NHWC>;
Dbias_op_gpu.cu.cc88 if (data_format == FORMAT_NHWC) { in compute()
227 if (data_format == FORMAT_NHWC) { in compute()
232 if (data_format == FORMAT_NHWC) { in compute()
253 if (data_format == FORMAT_NHWC) { in compute()
Dconv_ops_3d.cc63 OP_REQUIRES(context, data_format == FORMAT_NHWC, in launch()
254 strides[2] == 1 && data_format == FORMAT_NHWC) { in launch()
280 padding == Padding::VALID && data_format == FORMAT_NHWC) { in launch()
348 (compute_in_nhwc && data_format == FORMAT_NHWC) ? FORMAT_NHWC in launch()
355 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in launch()
392 compute_data_format == FORMAT_NHWC ? kComputeInNHWC : kComputeInNCHW; in launch()
633 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in launch()
Dconv_ops.cc85 CHECK(data_format == FORMAT_NHWC) << "Generic conv implementation only " in operator ()()
151 if (data_format != FORMAT_NHWC) { in operator ()()
188 if (data_format != FORMAT_NHWC) { in operator ()()
244 if (data_format != FORMAT_NHWC || dilation_rows != 1 || in Run()
680 col_stride == 1 && data_format == FORMAT_NHWC && in operator ()()
708 data_format == FORMAT_NHWC) { in operator ()()
751 (compute_in_nhwc && data_format == FORMAT_NHWC) ? FORMAT_NHWC in operator ()()
837 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
877 compute_data_format == FORMAT_NHWC ? kComputeInNHWC : kComputeInNCHW; in operator ()()
931 } else if (compute_data_format == FORMAT_NHWC) { in operator ()()
[all …]
Dconv_grad_filter_ops.cc307 OP_REQUIRES(context, data_format_ == FORMAT_NHWC, in Conv2DBackpropFilterOp()
389 OP_REQUIRES(context, data_format_ == FORMAT_NHWC, in Conv2DCustomBackpropFilterOp()
718 data_format == FORMAT_NHWC && (padding == VALID || padding == SAME)) { in operator ()()
758 data_format == FORMAT_NHWC) { in operator ()()
834 (compute_in_nhwc && data_format == FORMAT_NHWC) ? FORMAT_NHWC in operator ()()
852 compute_data_format == FORMAT_NHWC ? kComputeInNHWC : kComputeInNCHW; in operator ()()
899 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
920 if (data_format == FORMAT_NHWC && compute_data_format == FORMAT_NCHW) { in operator ()()
Dmaxpooling_op.cc229 context, data_format_ == FORMAT_NHWC, in MaxPoolingGradOp()
319 FORMAT_NHWC, in Compute()
462 context, data_format_ == FORMAT_NHWC, in MaxPoolingGradGradOp()
535 FORMAT_NHWC, in Compute()
775 context, data_format_ == FORMAT_NHWC, in MaxPoolingNoMaskOp()
837 context, data_format_ == FORMAT_NHWC, in MaxPoolingNoMaskV2Op()
956 FORMAT_NHWC, in Compute()
999 GetTensorDim(grad_out->shape(), FORMAT_NHWC, 'N'); in launch()
1030 const int64 batch_size = GetTensorDim(grad_out->shape(), FORMAT_NHWC, 'N'); in launch()
1076 FORMAT_NHWC, in Compute()
[all …]
Dlrn_op.cc243 ShapeFromFormat(FORMAT_NCHW, in.shape(), FORMAT_NHWC), in launch()
253 ShapeFromFormat(FORMAT_NCHW, output->shape(), FORMAT_NHWC), in launch()
537 FORMAT_NHWC), in launch()
547 FORMAT_NHWC), in launch()
557 FORMAT_NHWC), in launch()
567 ShapeFromFormat(FORMAT_NCHW, output->shape(), FORMAT_NHWC), in launch()
/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()
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h39 FORMAT_NHWC = 0, enumerator
118 case FORMAT_NHWC: in GetTensorSpatialDims()
147 case FORMAT_NHWC: in GetTensorDimsFromSpatialDims()
175 case FORMAT_NHWC: in GetTensorBatchDimIndex()
195 case FORMAT_NHWC: in GetTensorFeatureDimIndex()
232 case FORMAT_NHWC: in GetTensorSpatialDimIndex()
314 if (format == FORMAT_NHWC || format == FORMAT_NHWC_VECT_W) { in GetTensorDimIndex()
Dtensor_format.cc42 case FORMAT_NHWC: in ToString()
78 *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.cc680 case FORMAT_NHWC: in ImageDims()
1150 ImageDims(FORMAT_NHWC, batch, feature, input_dims)) in TEST_F()
1152 .Attr("ksize", ImageDims(FORMAT_NHWC, 1, 1, kernel_dims)) in TEST_F()
1153 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, stride_dims)) in TEST_F()
1165 AsInt32s(ImageDims(FORMAT_NHWC, batch, features, d.input_dims)); in TEST_F()
1167 ImageDims(FORMAT_NHWC, batch, features, d.output_dims); in TEST_F()
1173 .Attr("ksize", ImageDims(FORMAT_NHWC, 1, 1, d.kernel_dims)) in TEST_F()
1174 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, d.stride_dims)) in TEST_F()
1185 AsInt32s(ImageDims(FORMAT_NHWC, batch, features, d.input_dims)); in TEST_F()
1187 ImageDims(FORMAT_NHWC, batch, features, d.output_dims); in TEST_F()
[all …]

123