| /external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
| fused_conv2d_bias_activation_op.cc | 54 string data_format_str, filter_format_str;
55 TF_RETURN_IF_ERROR(c->GetAttr("data_format", &data_format_str));
59 FormatFromString(data_format_str, &data_format);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| batch_norm_op.cc | 31 string data_format_str; local
32 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str));
34 ctx, FormatFromString(data_format_str, &data_format_),
35 errors::InvalidArgument("Invalid data format: ", data_format_str));
108 string data_format_str; local
109 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str));
111 ctx, FormatFromString(data_format_str, &data_format_),
112 errors::InvalidArgument("Invalid data format: ", data_format_str));
|
| depthtospace_op.cc | 27 string data_format_str; local
28 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str));
29 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_),
|
| spacetodepth_op.cc | 27 string data_format_str; local
28 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str));
29 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_),
|
| pooling_ops.cc | 155 string data_format_str; local
156 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str));
157 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_),
254 string data_format_str; local
255 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str));
256 OP_REQUIRES(ctx, FormatFromString(data_format_str, &data_format_),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_tfconv_op.h | 53 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str));
59 ConvertMklToTf(this, context, data_format_str, op_data_type, has_avx512f_,
66 string data_format_str, DataType op_data_type,
130 string data_format_str, DataType op_data_type,
182 string data_format_str; member in class:tensorflow::MklToTfOp
|
| mkl_input_conversion_op.cc | 68 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str));
130 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str,
132 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str,
225 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str,
243 string data_format_str; member in class:tensorflow::MklInputConversionOp
259 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str));
363 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str,
365 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str,
460 MklToTfOp<Device, T>::ConvertMklToTf(this, context, data_format_str,
478 string data_format_str; member in class:tensorflow::MklInputConversionOp
[all...] |
| depthtospace_op.cc | 47 string data_format_str; local
48 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str));
49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_),
61 "Only NHWC data_format supported on CPU. Got ", data_format_str));
|
| spacetodepth_op.cc | 47 string data_format_str; local
48 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str));
49 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_),
61 "Only NHWC data_format supported on CPU. Got ", data_format_str));
|
| mkl_conv_ops.h | 336 string data_format_str; local
337 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str));
338 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_),
|
| maxpooling_op.cc | 929 string data_format_str; local
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| common_shape_fns.cc | 372 string data_format_str, filter_format_str; local
373 if (!c->GetAttr("data_format", &data_format_str).ok()) {
374 data_format_str = "NHWC";
381 if (!FormatFromString(data_format_str, &data_format)) {
383 data_format_str);
418 return errors::InvalidArgument("Conv2D on data format ", data_format_str,
628 string data_format_str; local
630 Status s = c->GetAttr("data_format", &data_format_str);
632 FormatFromString(data_format_str, &data_format);
778 string data_format_str; local
848 string data_format_str; local
[all...] |
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.cc | 91 string data_format_str, filter_format_str; local
93 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format_str));
94 OP_REQUIRES(context, FormatFromString(data_format_str, &data_format_),
|
| /external/tensorflow/tensorflow/core/graph/ |
| mkl_layout_pass.cc | 609 string data_format_str; local
614 CHECK_EQ(GetNodeAttr(n->def(), "data_format", &data_format_str).ok(), true);
615 CHECK_EQ(FormatFromString(data_format_str, &data_format), true);
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| array_ops.cc | [all...] |
