| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | permute_op.cc | 38 OP_REQUIRES(ctx, FormatFromString(src_format_, &data_format), in DataFormatVecPermuteOp()
44 OP_REQUIRES(ctx, FormatFromString(dst_format_, &data_format), in DataFormatVecPermuteOp()
|
| D | bias_ops.cc | 34 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in BiasOp()
83 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in BiasAddGradOp()
|
| D | pooling_ops.cc | 157 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in MaxPoolOp()
210 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in AvgPoolOp()
358 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool2DGradOp()
397 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in AvgPoolGradOp()
631 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool2DGradGradOp()
649 OP_REQUIRES(ctx, FormatFromString(data_format, &data_format_), in MaxPool3DGradGradOp()
|
| D | batch_norm_op.cc | 37 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormOp()
120 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormGradOp()
|
| D | depthtospace_op.cc | 31 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
|
| D | spacetodepth_op.cc | 31 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 55 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingOp()
123 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingOp()
221 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOp()
383 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOp()
463 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in AvgPoolingGradOpCustomGPUKernel()
|
| D | bias_op.cc | 93 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasOp()
238 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasGradOp()
336 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasOp()
483 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in BiasGradOp()
|
| D | maxpooling_op.cc | 217 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradOp()
354 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradOp()
456 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradGradOp()
661 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingGradGradOp()
759 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskOp()
812 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskV2Op()
1013 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in MaxPoolingGradWithArgmaxOp()
1124 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskOp()
1215 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingNoMaskV2Op()
|
| D | pooling_ops_3d.cc | 136 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()
|
| D | depthtospace_op.cc | 49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
|
| D | spacetodepth_op.cc | 49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
|
| D | conv_grad_ops_3d.cc | 175 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropInputOp()
281 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DCustomBackpropInputOp()
631 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropFilterOp()
743 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DCustomBackpropFilterOp()
1100 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropInputOp()
1483 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv3DBackpropFilterOp()
|
| D | mkl_conv_grad_bias_ops.cc | 59 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklConv2DCustomBackpropBiasOp()
|
| D | mkl_avgpooling_op.cc | 51 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklAvgPoolingOp()
227 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklAvgPoolingGradOp()
|
| D | pooling_ops_common.h | 85 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MaxPoolingOp()
295 OP_REQUIRES(context, FormatFromString(data_format, &data_format_),
|
| D | mkl_maxpooling_op.cc | 53 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklMaxPoolingOp()
212 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in MklMaxPoolingGradOp()
|
| D | fused_batch_norm_op.cc | 532 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in FusedBatchNormOp()
609 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), in FusedBatchNormGradOp()
|
| D | conv_ops_using_gemm.cc | 436 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in Conv2DUsingGemmOp()
|
| D | depthwise_conv_op.cc | 271 OP_REQUIRES(context, FormatFromString(data_format, &data_format_), in DepthwiseConv2dNativeOp()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.cc | 74 bool FormatFromString(const string& format_str, TensorFormat* format) { in FormatFromString() function
|
| D | tensor_format_test.cc | 153 EXPECT_TRUE(FormatFromString(expected_format_str, &format)); in TEST()
|
| D | tensor_format.h | 96 bool FormatFromString(const string& format_str, TensorFormat* format);
|
| /external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
| D | fused_conv2d_bias_activation_op.cc | 59 FormatFromString(data_format_str, &data_format); in __anon4e6e467d0102()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 421 if (!FormatFromString(data_format_str, &data_format)) { in Conv2DShapeImpl()
725 FormatFromString(data_format_str, &data_format); in AvgPoolShape()
799 if (!FormatFromString(data_format_str, &data_format)) { in FusedBatchNormShape()
835 if (!FormatFromString(data_format_str, &data_format)) { in FusedBatchNormGradShape()
875 FormatFromString(data_format_str, &data_format); in MaxPoolShape()
945 FormatFromString(data_format_str, &data_format); in MaxPoolV2Shape()
|