Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc34 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in BiasOp()
83 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in BiasAddGradOp()
Dpooling_ops.cc159 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in MaxPoolOp()
218 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in AvgPoolOp()
379 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool2DGradOp()
418 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in AvgPoolGradOp()
654 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool2DGradGradOp()
672 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool3DGradGradOp()
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()
Dbatch_norm_op.cc45 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormOp()
246 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormGradOp()
Ddepthtospace_op.cc31 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
Dspacetodepth_op.cc31 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
Dconv_op_helpers.cc176 if (!FormatFromString(data_format, &attrs.data_format)) { in Create()
/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op.cc61 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingOp()
139 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingOp()
255 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOp()
420 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOp()
507 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOpCustomGPUKernel()
Dbias_op.cc92 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasOp()
223 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasGradOp()
303 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasOp()
451 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasGradOp()
Dmaxpooling_op.cc226 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradOp()
352 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradOp()
459 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradGradOp()
669 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradGradOp()
772 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskOp()
834 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskV2Op()
1046 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in MaxPoolingGradWithArgmaxOp()
1167 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskOp()
1266 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskV2Op()
Ddepthtospace_op.cc49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
Dpooling_ops_3d.cc131 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Pooling3DOp()
305 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPooling3dGradOp()
459 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPooling3dGradOp()
658 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPooling3dGradGradOp()
Dspacetodepth_op.cc64 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
Dconv_grad_ops_3d.cc184 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropInputOp()
291 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DCustomBackpropInputOp()
642 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropFilterOp()
754 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DCustomBackpropFilterOp()
1113 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropInputOp()
1594 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropFilterOp()
Dpooling_ops_common.h101 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingOp()
340 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
Dconv_grad_input_ops.h359 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
480 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
Dconv_ops_using_gemm.cc438 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DUsingGemmOp()
Ddepthwise_conv_op.cc276 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in DepthwiseConv2dNativeOp()
Dconv_grad_filter_ops.cc267 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DBackpropFilterOp()
387 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DCustomBackpropFilterOp()
/external/tensorflow/tensorflow/core/util/
Dtensor_format.cc76 bool FormatFromString(absl::string_view format_str, TensorFormat* format) { in FormatFromString() function
Dtensor_format_test.cc153 EXPECT_TRUE(FormatFromString(expected_format_str, &format)); in TEST()
Dtensor_format.h101 bool FormatFromString(absl::string_view format_str, TensorFormat* format);
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc588 if (!FormatFromString(data_format_str, &data_format)) { in Conv2DShapeImpl()
855 if (!FormatFromString(data_format_str, &data_format)) { in Conv2DBackpropInputShape()
955 if (!s.ok() || !FormatFromString(data_format_str, &data_format)) { in DepthwiseConv2DNativeShapeImpl()
1059 FormatFromString(data_format_str, &data_format); in AvgPoolShape()
1135 if (!FormatFromString(data_format_str, &data_format)) { in FusedBatchNormShape()
1185 if (!FormatFromString(data_format_str, &data_format)) { in FusedBatchNormExShape()
1208 if (!FormatFromString(data_format_str, &data_format)) { in FusedBatchNormGradShape()
1500 FormatFromString(data_format_str, &data_format); in MaxPoolShapeImpl()
1607 FormatFromString(data_format_str, &data_format); in MaxPoolV2Shape()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc453 bool is_valid = FormatFromString(data_format, &format); in Verify()
521 bool is_valid = FormatFromString(data_format, &format); in Verify()
1524 auto is_valid = FormatFromString(data_format, &format); in Verify()
1584 auto data_format_is_valid = FormatFromString(data_format.str(), &format); in inferConvReturnTypes()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h566 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in MklConvBackpropCommonOp()

12