Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc33 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in BiasOp()
80 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in BiasAddGradOp()
Dpooling_ops.cc157 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in MaxPool2DOp()
256 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in AvgPool2DOp()
368 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool2DGradOp()
513 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in AvgPool2DGradOp()
Dbatch_norm_op.cc34 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormOp()
111 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormGradOp()
Dconv_ops.cc188 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in ConvOp()
339 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in ConvBackpropInputOp()
497 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in ConvBackpropFilterOp()
Ddepthtospace_op.cc29 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
Dspacetodepth_op.cc29 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op.cc55 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingOp()
120 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingOp()
214 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOp()
376 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOp()
450 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOpCustomGPUKernel()
Dbias_op.cc90 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasOp()
210 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasGradOp()
314 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasOp()
373 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasGradOp()
Dmaxpooling_op.cc203 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradOp()
340 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradOp()
442 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradGradOp()
647 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradGradOp()
745 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskOp()
798 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskV2Op()
932 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in MaxPoolingGradWithArgmaxOp()
1037 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskOp()
1122 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskV2Op()
Dpooling_ops_3d.cc136 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Pooling3DOp()
309 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPooling3dGradOp()
463 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPooling3dGradOp()
662 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPooling3dGradGradOp()
Ddepthtospace_op.cc49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
Dspacetodepth_op.cc49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
Dconv_grad_ops_3d.cc146 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropInputOp()
256 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropFilterOp()
438 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropInputOp()
782 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropFilterOp()
Dmkl_conv_grad_bias_ops.cc56 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklConv2DCustomBackpropBiasOp()
Dmkl_fused_batch_norm_op.cc56 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormOp()
353 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormGradOp()
698 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormOp()
1041 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in MklFusedBatchNormGradOp()
Dconv_grad_filter_ops.cc185 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DFastBackpropFilterOp()
298 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DCustomBackpropFilterOp()
543 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DSlowBackpropFilterOp()
Dmkl_avgpooling_op.cc51 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklAvgPoolingOp()
227 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklAvgPoolingGradOp()
Dpooling_ops_common.h85 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingOp()
295 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
Dmkl_maxpooling_op.cc53 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklMaxPoolingOp()
211 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklMaxPoolingGradOp()
Dmkl_conv_ops.cc74 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklConv2DOp()
497 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklConv2DOp()
Dconv_grad_input_ops.cc188 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DFastBackpropInputOp()
303 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DCustomBackpropInputOp()
615 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DSlowBackpropInputOp()
/external/tensorflow/tensorflow/core/util/
Dtensor_format.cc64 bool FormatFromString(const string& format_str, TensorFormat* format) { in FormatFromString() function
Dtensor_format_test.cc138 EXPECT_TRUE(FormatFromString(expected_format_str, &format)); in TEST()
/external/tensorflow/tensorflow/contrib/fused_conv/ops/
Dfused_conv2d_bias_activation_op.cc59 FormatFromString(data_format_str, &data_format); in __anonea18cef10102()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc381 if (!FormatFromString(data_format_str, &data_format)) { in Conv2DShape()
632 FormatFromString(data_format_str, &data_format); in AvgPoolShape()
782 FormatFromString(data_format_str, &data_format); in MaxPoolShape()
852 FormatFromString(data_format_str, &data_format); in MaxPoolV2Shape()

12