/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | batch_norm_op.cc | 34 string data_format_str; in FusedBatchNormOp() local 35 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in FusedBatchNormOp() 37 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormOp() 38 errors::InvalidArgument("Invalid data format: ", data_format_str)); in FusedBatchNormOp() 117 string data_format_str; in FusedBatchNormGradOp() local 118 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in FusedBatchNormGradOp() 120 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormGradOp() 121 errors::InvalidArgument("Invalid data format: ", data_format_str)); in FusedBatchNormGradOp()
|
D | depthtospace_op.cc | 29 string data_format_str; in DepthToSpaceOp() local 30 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in DepthToSpaceOp() 31 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp()
|
D | spacetodepth_op.cc | 29 string data_format_str; in SpaceToDepthOp() local 30 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in SpaceToDepthOp() 31 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
|
D | pooling_ops.cc | 155 string data_format_str; in MaxPoolOp() local 156 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in MaxPoolOp() 157 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in MaxPoolOp() 208 string data_format_str; in AvgPoolOp() local 209 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in AvgPoolOp() 210 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in AvgPoolOp()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | mkl_tfconv_op.h | 50 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str)); in MklToTfOp() 56 ConvertMklToTf(this, context, data_format_str, op_data_type, has_avx512f_, in Compute() 62 string data_format_str, DataType op_data_type, in ConvertMklToTf() argument 123 string data_format_str;
|
D | mkl_input_conversion_op.cc | 64 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str)); in MklInputConversionOp() 173 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str, in Compute() 176 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str, in Compute() 266 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str, in Compute() 285 string data_format_str; member in tensorflow::MklInputConversionOp
|
D | depthtospace_op.cc | 47 string data_format_str; in DepthToSpaceOp() local 48 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str)); in DepthToSpaceOp() 49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in DepthToSpaceOp() 61 "Only NHWC data_format supported on CPU. Got ", data_format_str)); in DepthToSpaceOp()
|
D | spacetodepth_op.cc | 47 string data_format_str; in SpaceToDepthOp() local 48 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str)); in SpaceToDepthOp() 49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp() 61 "Only NHWC data_format supported on CPU. Got ", data_format_str)); in SpaceToDepthOp()
|
D | mkl_conv_ops.h | 555 string data_format_str; in MklConvBackpropCommonOp() local 556 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str)); in MklConvBackpropCommonOp() 557 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in MklConvBackpropCommonOp()
|
D | maxpooling_op.cc | 1010 string data_format_str; in MaxPoolingGradWithArgmaxOp() local 1011 auto status = context->GetAttr("data_format", &data_format_str); in MaxPoolingGradWithArgmaxOp() 1013 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in MaxPoolingGradWithArgmaxOp()
|
/external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
D | fused_conv2d_bias_activation_op.cc | 54 string data_format_str, filter_format_str; in __anon4e6e467d0102() local 55 TF_RETURN_IF_ERROR(c->GetAttr("data_format", &data_format_str)); in __anon4e6e467d0102() 59 FormatFromString(data_format_str, &data_format); in __anon4e6e467d0102()
|
/external/tensorflow/tensorflow/core/framework/ |
D | common_shape_fns.cc | 412 string data_format_str, filter_format_str; in Conv2DShapeImpl() local 413 if (!c->GetAttr("data_format", &data_format_str).ok()) { in Conv2DShapeImpl() 414 data_format_str = "NHWC"; in Conv2DShapeImpl() 421 if (!FormatFromString(data_format_str, &data_format)) { in Conv2DShapeImpl() 423 data_format_str); in Conv2DShapeImpl() 458 return errors::InvalidArgument("Conv2D on data format ", data_format_str, in Conv2DShapeImpl() 721 string data_format_str; in AvgPoolShape() local 723 Status s = c->GetAttr("data_format", &data_format_str); in AvgPoolShape() 725 FormatFromString(data_format_str, &data_format); in AvgPoolShape() 796 string data_format_str; in FusedBatchNormShape() local [all …]
|
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
D | fused_conv2d_bias_activation_op.cc | 98 string data_format_str, filter_format_str; in FusedConv2DBiasActivationOp() local 100 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str)); in FusedConv2DBiasActivationOp() 101 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in FusedConv2DBiasActivationOp()
|
/external/tensorflow/tensorflow/core/ops/ |
D | array_ops.cc | 2318 string data_format_str; in __anon7c94107b3702() local 2319 TF_RETURN_IF_ERROR(c->GetAttr("data_format", &data_format_str)); in __anon7c94107b3702() 2321 FormatFromString(data_format_str, &data_format); in __anon7c94107b3702() 2372 string data_format_str; in __anon7c94107b3802() local 2373 TF_RETURN_IF_ERROR(c->GetAttr("data_format", &data_format_str)); in __anon7c94107b3802() 2375 FormatFromString(data_format_str, &data_format); in __anon7c94107b3802()
|
/external/tensorflow/tensorflow/core/graph/ |
D | mkl_layout_pass.cc | 1197 string data_format_str; in NonDepthBatchWisePoolRewrite() local 1202 CHECK_EQ(GetNodeAttr(n->def(), "data_format", &data_format_str).ok(), true); in NonDepthBatchWisePoolRewrite() 1203 CHECK_EQ(FormatFromString(data_format_str, &data_format), true); in NonDepthBatchWisePoolRewrite()
|