Home
last modified time | relevance | path

Searched refs:data_format_str (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc34 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()
Ddepthtospace_op.cc29 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()
Dspacetodepth_op.cc29 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()
Dpooling_ops.cc155 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/
Dmkl_tfconv_op.h50 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;
Dmkl_input_conversion_op.cc64 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
Ddepthtospace_op.cc47 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()
Dspacetodepth_op.cc47 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()
Dmkl_conv_ops.h555 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()
Dmaxpooling_op.cc1010 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/
Dfused_conv2d_bias_activation_op.cc54 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/
Dcommon_shape_fns.cc412 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/
Dfused_conv2d_bias_activation_op.cc98 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/
Darray_ops.cc2318 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/
Dmkl_layout_pass.cc1197 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()