| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_tfconv_op.h | 52 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str)); in MklToTfOp()
58 ConvertMklToTf(this, context, data_format_str, op_data_type, has_avx512f_, in Compute()
65 string data_format_str, DataType op_data_type, in ConvertMklToTf() argument
131 string data_format_str;
|
| D | mkl_input_conversion_op.cc | 62 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str)); in MklInputConversionOp()
175 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str, in Compute()
178 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str, in Compute()
270 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str, in Compute()
289 string data_format_str; member in tensorflow::MklInputConversionOp
|
| D | mkl_conv_ops.h | 564 string data_format_str; in MklConvBackpropCommonOp() local
565 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str)); in MklConvBackpropCommonOp()
566 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in MklConvBackpropCommonOp()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 42 string data_format_str; in FusedBatchNormOp() local
43 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in FusedBatchNormOp()
45 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormOp()
46 errors::InvalidArgument("Invalid data format: ", data_format_str)); in FusedBatchNormOp()
243 string data_format_str; in FusedBatchNormGradOp() local
244 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in FusedBatchNormGradOp()
246 ctx, FormatFromString(data_format_str, &data_format_), in FusedBatchNormGradOp()
247 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 | 157 string data_format_str; in MaxPoolOp() local
158 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in MaxPoolOp()
159 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in MaxPoolOp()
216 string data_format_str; in AvgPoolOp() local
217 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in AvgPoolOp()
218 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_), in AvgPoolOp()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 579 string data_format_str, filter_format_str; in Conv2DShapeImpl() local
580 if (!c->GetAttr("data_format", &data_format_str).ok()) { in Conv2DShapeImpl()
581 data_format_str = "NHWC"; in Conv2DShapeImpl()
588 if (!FormatFromString(data_format_str, &data_format)) { in Conv2DShapeImpl()
590 data_format_str); in Conv2DShapeImpl()
625 return errors::InvalidArgument("Conv2D on data format ", data_format_str, in Conv2DShapeImpl()
850 string data_format_str; in Conv2DBackpropInputShape() local
851 if (!c->GetAttr("data_format", &data_format_str).ok()) { in Conv2DBackpropInputShape()
852 data_format_str = "NHWC"; in Conv2DBackpropInputShape()
855 if (!FormatFromString(data_format_str, &data_format)) { in Conv2DBackpropInputShape()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 | 62 string data_format_str; in SpaceToDepthOp() local
63 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str)); in SpaceToDepthOp()
64 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in SpaceToDepthOp()
76 "Only NHWC data_format supported on CPU. Got ", data_format_str)); in SpaceToDepthOp()
|
| D | maxpooling_op.cc | 1043 string data_format_str; in MaxPoolingGradWithArgmaxOp() local
1044 auto status = context->GetAttr("data_format", &data_format_str); in MaxPoolingGradWithArgmaxOp()
1046 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_), in MaxPoolingGradWithArgmaxOp()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2475 string data_format_str; in __anondb9326b23702() local
2476 TF_RETURN_IF_ERROR(c->GetAttr("data_format", &data_format_str)); in __anondb9326b23702()
2478 FormatFromString(data_format_str, &data_format); in __anondb9326b23702()
2529 string data_format_str; in __anondb9326b23802() local
2530 TF_RETURN_IF_ERROR(c->GetAttr("data_format", &data_format_str)); in __anondb9326b23802()
2532 FormatFromString(data_format_str, &data_format); in __anondb9326b23802()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1568 string data_format_str; in NonDepthBatchWisePoolRewrite() local
1573 TF_CHECK_OK(GetNodeAttr(n->def(), "data_format", &data_format_str)); in NonDepthBatchWisePoolRewrite()
1574 bool result = FormatFromString(data_format_str, &data_format); in NonDepthBatchWisePoolRewrite()
|