Home
last modified time | relevance | path

Searched refs:FormatFromString (Results 1 – 25 of 39) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpooling_ops.cc167 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in MaxPoolOp()
261 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in AvgPoolOp()
427 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool2DGradOp()
464 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in AvgPoolGradOp()
706 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool2DGradGradOp()
724 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool3DGradGradOp()
H A Ddata_format_ops.cc97 OP_REQUIRES(ctx, FormatFromString(src_format_, &data_format), in DataFormatVecPermuteOp()
103 OP_REQUIRES(ctx, FormatFromString(dst_format_, &data_format), in DataFormatVecPermuteOp()
H A Dbatch_norm_op.cc49 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormOp()
250 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormGradOp()
H A Dbias_ops.cc35 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in BiasOp()
H A Ddepthtospace_op.cc31 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
H A Dspacetodepth_op.cc31 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
H A Dconv_op_helpers.cc175 if (!FormatFromString(data_format, &attrs.data_format)) { in Create()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Davgpooling_op.cc62 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingOp()
140 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingOp()
266 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOp()
432 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOp()
519 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOpCustomGPUKernel()
H A Dbias_op.cc95 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasOp()
165 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasGradOp()
244 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasOp()
392 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasGradOp()
H A Dmaxpooling_op.cc230 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradOp()
364 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradOp()
471 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradGradOp()
693 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradGradOp()
807 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskOp()
869 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskV2Op()
1095 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in MaxPoolingGradWithArgmaxOp()
1234 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskOp()
1340 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskV2Op()
H A Dpooling_ops_3d.cc131 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Pooling3DOp()
306 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPooling3dGradOp()
482 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPooling3dGradOp()
681 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPooling3dGradGradOp()
H A Dconv_grad_ops_3d.cc188 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropInputOp()
317 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DCustomBackpropInputOp()
695 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropFilterOp()
833 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DCustomBackpropFilterOp()
1225 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropInputOp()
1604 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropFilterOp()
H A Ddepthtospace_op.cc49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
H A Dpooling_ops_common.h96 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingOp()
339 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
H A Dspacetodepth_op.cc64 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
H A Dconv_grad_input_ops.h360 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
495 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
H A Dconv_grad_filter_ops.cc271 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DBackpropFilterOp()
391 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DCustomBackpropFilterOp()
H A Dconv_ops_using_gemm.cc438 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DUsingGemmOp()
H A Ddepthwise_conv_op.cc280 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in DepthwiseConv2dNativeOp()
H A Ddepthwise_conv_grad_op.cc565 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in DepthwiseConv2dNativeBackpropInputOp()
1068 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in DepthwiseConv2dNativeBackpropFilterOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc596 if (!FormatFromString(data_format_str, &data_format)) { in Conv2DShapeImpl()
891 if (!FormatFromString(data_format_str, &data_format)) { in Conv2DBackpropInputShape()
1010 if (!s.ok() || !FormatFromString(data_format_str, &data_format)) { in DepthwiseConv2DNativeShapeImpl()
1114 FormatFromString(data_format_str, &data_format); in AvgPoolShape()
1190 if (!FormatFromString(data_format_str, &data_format)) { in FusedBatchNormShape()
1240 if (!FormatFromString(data_format_str, &data_format)) { in FusedBatchNormExShape()
1263 if (!FormatFromString(data_format_str, &data_format)) { in FusedBatchNormGradShape()
1312 if (!FormatFromString(data_format_str, &data_format)) { in FusedBatchNormGradExShape()
1591 FormatFromString(data_format_str, &data_format); in MaxPoolShapeImpl()
1698 FormatFromString(data_format_str, &data_format); in MaxPoolV2Shape()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_format.cc76 bool FormatFromString(absl::string_view format_str, TensorFormat* format) { in FormatFromString() function
H A Dtensor_format_test.cc153 EXPECT_TRUE(FormatFromString(expected_format_str, &format)); in TEST()
H A Dtensor_format.h101 bool FormatFromString(absl::string_view format_str, TensorFormat* format);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_a_m.cc503 bool is_valid = FormatFromString(data_format, &format); in verify()
565 bool is_valid = FormatFromString(data_format, &format); in verify()
1617 auto data_format_is_valid = FormatFromString(data_format.str(), &format); in Verify()
1679 auto is_valid = FormatFromString(data_format, &format); in Verify()
1741 auto data_format_is_valid = FormatFromString(data_format.str(), &format); in inferConvReturnTypeComponents()

12