Home
last modified time | relevance | path

Searched refs:GetAttr (Results 1 – 25 of 392) sorted by relevance

12345678910>>...16

/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.h158 TF_RETURN_IF_ERROR(ctx->GetAttr("sparse_types", &sparse_types)); in Init()
159 TF_RETURN_IF_ERROR(ctx->GetAttr("Ndense", &num_dense)); in Init()
160 TF_RETURN_IF_ERROR(ctx->GetAttr("Nsparse", &num_sparse)); in Init()
161 TF_RETURN_IF_ERROR(ctx->GetAttr("Tdense", &dense_types)); in Init()
162 TF_RETURN_IF_ERROR(ctx->GetAttr("dense_shapes", &dense_shapes)); in Init()
215 TF_RETURN_IF_ERROR(ctx->GetAttr("sparse_keys", &sparse_keys)); in Init()
216 TF_RETURN_IF_ERROR(ctx->GetAttr("sparse_types", &sparse_types)); in Init()
217 TF_RETURN_IF_ERROR(ctx->GetAttr("dense_keys", &dense_keys)); in Init()
218 TF_RETURN_IF_ERROR(ctx->GetAttr("Tdense", &dense_types)); in Init()
219 TF_RETURN_IF_ERROR(ctx->GetAttr("dense_shapes", &dense_shapes)); in Init()
[all …]
/external/tensorflow/tensorflow/lite/experimental/microfrontend/ops/
Daudio_microfrontend_op.cc70 TF_RETURN_IF_ERROR(ctx->GetAttr("sample_rate", &sample_rate)); in __anon68ee72bf0102()
72 TF_RETURN_IF_ERROR(ctx->GetAttr("window_size", &window_size)); in __anon68ee72bf0102()
75 TF_RETURN_IF_ERROR(ctx->GetAttr("window_step", &window_step)); in __anon68ee72bf0102()
79 TF_RETURN_IF_ERROR(ctx->GetAttr("num_channels", &num_channels)); in __anon68ee72bf0102()
81 TF_RETURN_IF_ERROR(ctx->GetAttr("left_context", &left_context)); in __anon68ee72bf0102()
83 TF_RETURN_IF_ERROR(ctx->GetAttr("right_context", &right_context)); in __anon68ee72bf0102()
85 TF_RETURN_IF_ERROR(ctx->GetAttr("frame_stride", &frame_stride)); in __anon68ee72bf0102()
156 OP_REQUIRES_OK(ctx, ctx->GetAttr("sample_rate", &sample_rate_)); in AudioMicrofrontendOp()
159 OP_REQUIRES_OK(ctx, ctx->GetAttr("window_size", &window_size)); in AudioMicrofrontendOp()
163 OP_REQUIRES_OK(ctx, ctx->GetAttr("window_step", &window_step)); in AudioMicrofrontendOp()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc76 OP_REQUIRES_OK(c, c->GetAttr("group_size", &col_params_.group.group_size)); in CollectiveGatherOpKernel()
77 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_.group.group_key)); in CollectiveGatherOpKernel()
79 c, c->GetAttr("instance_key", &col_params_.instance.instance_key)); in CollectiveGatherOpKernel()
80 OP_REQUIRES_OK(c, c->GetAttr("T", &col_params_.instance.data_type)); in CollectiveGatherOpKernel()
81 OP_REQUIRES_OK(c, c->GetAttr("shape", &col_params_.instance.shape)); in CollectiveGatherOpKernel()
127 OP_REQUIRES_OK(c, c->GetAttr("group_size", &col_params_.group.group_size)); in CollectiveReduceOpKernel()
128 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_.group.group_key)); in CollectiveReduceOpKernel()
130 c, c->GetAttr("instance_key", &col_params_.instance.instance_key)); in CollectiveReduceOpKernel()
132 c, c->GetAttr("subdiv_offsets", in CollectiveReduceOpKernel()
135 OP_REQUIRES_OK(c, c->GetAttr("merge_op", &merge_op_name)); in CollectiveReduceOpKernel()
[all …]
Dencode_jpeg_op.cc35 OP_REQUIRES_OK(context, context->GetAttr("format", &format_)); in EncodeJpegOp()
48 OP_REQUIRES_OK(context, context->GetAttr("quality", &flags_.quality)); in EncodeJpegOp()
53 context->GetAttr("progressive", &flags_.progressive)); in EncodeJpegOp()
55 context, context->GetAttr("optimize_size", &flags_.optimize_jpeg_size)); in EncodeJpegOp()
56 OP_REQUIRES_OK(context, context->GetAttr("chroma_downsampling", in EncodeJpegOp()
60 OP_REQUIRES_OK(context, context->GetAttr("density_unit", &density_unit)); in EncodeJpegOp()
71 OP_REQUIRES_OK(context, context->GetAttr("x_density", &flags_.x_density)); in EncodeJpegOp()
72 OP_REQUIRES_OK(context, context->GetAttr("y_density", &flags_.y_density)); in EncodeJpegOp()
73 OP_REQUIRES_OK(context, context->GetAttr("xmp_metadata", &xmp_metadata_)); in EncodeJpegOp()
Dscoped_allocator_ops.cc31 OP_REQUIRES_OK(context, context->GetAttr("T", &dtype_)); in ScopedAllocatorOp()
32 OP_REQUIRES_OK(context, context->GetAttr("shapes", &shapes_)); in ScopedAllocatorOp()
33 OP_REQUIRES_OK(context, context->GetAttr("sa_name", &name_)); in ScopedAllocatorOp()
34 OP_REQUIRES_OK(context, context->GetAttr("id", &id_)); in ScopedAllocatorOp()
35 OP_REQUIRES_OK(context, context->GetAttr("expected_call_count", in ScopedAllocatorOp()
95 OP_REQUIRES_OK(context, context->GetAttr("shape", &shape_)); in ScopedAllocatorConcatOp()
96 OP_REQUIRES_OK(context, context->GetAttr("T", &dtype_)); in ScopedAllocatorConcatOp()
97 OP_REQUIRES_OK(context, context->GetAttr("reshape", &reshape_)); in ScopedAllocatorConcatOp()
99 OP_REQUIRES_OK(context, context->GetAttr("sa_name", &name_)); in ScopedAllocatorConcatOp()
100 OP_REQUIRES_OK(context, context->GetAttr("id", &id_)); in ScopedAllocatorConcatOp()
[all …]
Dcandidate_sampler_ops.cc36 OP_REQUIRES_OK(context, context->GetAttr("num_sampled", &num_sampled_)); in BaseCandidateSamplerOp()
37 OP_REQUIRES_OK(context, context->GetAttr("num_true", &num_true_)); in BaseCandidateSamplerOp()
38 OP_REQUIRES_OK(context, context->GetAttr("unique", &unique_)); in BaseCandidateSamplerOp()
118 OP_REQUIRES_OK(context, context->GetAttr("range_max", &range_max)); in SimpleCandidateSamplerOp()
142 OP_REQUIRES_OK(context, context->GetAttr("num_sampled", &range_max)); in AllCandidateSamplerOp()
155 OP_REQUIRES_OK(context, context->GetAttr("range_max", &range_max)); in FixedUnigramCandidateSamplerOp()
157 OP_REQUIRES_OK(context, context->GetAttr("vocab_file", &vocab_file)); in FixedUnigramCandidateSamplerOp()
159 OP_REQUIRES_OK(context, context->GetAttr("unigrams", &unigrams)); in FixedUnigramCandidateSamplerOp()
167 OP_REQUIRES_OK(context, context->GetAttr("distortion", &distortion)); in FixedUnigramCandidateSamplerOp()
170 context->GetAttr("num_reserved_ids", &num_reserved_ids)); in FixedUnigramCandidateSamplerOp()
[all …]
Dsendrecv_ops.cc57 OP_REQUIRES_OK(ctx, ctx->GetAttr("send_device", &send_device)); in SendOp()
59 OP_REQUIRES_OK(ctx, ctx->GetAttr("recv_device", &recv_device)); in SendOp()
62 ctx, ctx->GetAttr("send_device_incarnation", in SendOp()
65 OP_REQUIRES_OK(ctx, ctx->GetAttr("tensor_name", &tensor_name)); in SendOp()
72 if (!ctx->GetAttr("_hostmem_sendrecv", &hostmem_sendrecv_).ok()) { in SendOp()
125 OP_REQUIRES_OK(ctx, ctx->GetAttr("send_device", &send_device)); in RecvOp()
127 OP_REQUIRES_OK(ctx, ctx->GetAttr("recv_device", &recv_device)); in RecvOp()
130 ctx, ctx->GetAttr("send_device_incarnation", in RecvOp()
133 OP_REQUIRES_OK(ctx, ctx->GetAttr("tensor_name", &tensor_name)); in RecvOp()
140 if (!ctx->GetAttr("_hostmem_sendrecv", &hostmem_sendrecv_).ok()) { in RecvOp()
Dfake_quant_ops.cc61 OP_REQUIRES_OK(context, context->GetAttr("min", &min_)); in FakeQuantWithMinMaxArgsOp()
62 OP_REQUIRES_OK(context, context->GetAttr("max", &max_)); in FakeQuantWithMinMaxArgsOp()
67 OP_REQUIRES_OK(context, context->GetAttr("num_bits", &num_bits)); in FakeQuantWithMinMaxArgsOp()
72 OP_REQUIRES_OK(context, context->GetAttr("narrow_range", &narrow_range)); in FakeQuantWithMinMaxArgsOp()
101 OP_REQUIRES_OK(context, context->GetAttr("min", &min_)); in FakeQuantWithMinMaxArgsGradientOp()
102 OP_REQUIRES_OK(context, context->GetAttr("max", &max_)); in FakeQuantWithMinMaxArgsGradientOp()
107 OP_REQUIRES_OK(context, context->GetAttr("num_bits", &num_bits)); in FakeQuantWithMinMaxArgsGradientOp()
112 OP_REQUIRES_OK(context, context->GetAttr("narrow_range", &narrow_range)); in FakeQuantWithMinMaxArgsGradientOp()
179 OP_REQUIRES_OK(context, context->GetAttr("num_bits", &num_bits)); in FakeQuantWithMinMaxVarsOp()
184 OP_REQUIRES_OK(context, context->GetAttr("narrow_range", &narrow_range)); in FakeQuantWithMinMaxVarsOp()
[all …]
Dquantize_and_dequantize_op.cc46 OP_REQUIRES_OK(ctx, ctx->GetAttr("signed_input", &signed_input_)); in QuantizeAndDequantizeV2Op()
47 OP_REQUIRES_OK(ctx, ctx->GetAttr("num_bits", &num_bits_)); in QuantizeAndDequantizeV2Op()
51 OP_REQUIRES_OK(ctx, ctx->GetAttr("range_given", &range_given_)); in QuantizeAndDequantizeV2Op()
54 OP_REQUIRES_OK(ctx, ctx->GetAttr("round_mode", &round_mode_string)); in QuantizeAndDequantizeV2Op()
117 OP_REQUIRES_OK(ctx, ctx->GetAttr("signed_input", &signed_input_)); in QuantizeAndDequantizeV3Op()
118 OP_REQUIRES_OK(ctx, ctx->GetAttr("range_given", &range_given_)); in QuantizeAndDequantizeV3Op()
169 OP_REQUIRES_OK(ctx, ctx->GetAttr("signed_input", &signed_input_)); in QuantizeAndDequantizeOp()
170 OP_REQUIRES_OK(ctx, ctx->GetAttr("num_bits", &num_bits_)); in QuantizeAndDequantizeOp()
174 OP_REQUIRES_OK(ctx, ctx->GetAttr("range_given", &range_given_)); in QuantizeAndDequantizeOp()
175 OP_REQUIRES_OK(ctx, ctx->GetAttr("input_min", &input_min_)); in QuantizeAndDequantizeOp()
[all …]
Davgpooling_op.cc54 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in AvgPoolingOp()
62 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in AvgPoolingOp()
66 OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_)); in AvgPoolingOp()
70 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in AvgPoolingOp()
122 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in AvgPoolingOp()
125 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in AvgPoolingOp()
129 OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_)); in AvgPoolingOp()
133 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in AvgPoolingOp()
220 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in AvgPoolingGradOp()
228 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in AvgPoolingGradOp()
[all …]
Dstrided_slice_op.cc83 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceOp()
84 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_mask)); in StridedSliceOp()
85 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceOp()
86 OP_REQUIRES_OK(context, context->GetAttr("new_axis_mask", &new_axis_mask)); in StridedSliceOp()
88 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceOp()
190 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceGradOp()
191 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_mask)); in StridedSliceGradOp()
192 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceGradOp()
193 OP_REQUIRES_OK(context, context->GetAttr("new_axis_mask", &new_axis_mask)); in StridedSliceGradOp()
195 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceGradOp()
[all …]
Dmaxpooling_op.cc216 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in MaxPoolingGradOp()
226 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in MaxPoolingGradOp()
230 OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_)); in MaxPoolingGradOp()
243 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in MaxPoolingGradOp()
353 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in MaxPoolingGradOp()
357 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in MaxPoolingGradOp()
361 OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_)); in MaxPoolingGradOp()
371 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in MaxPoolingGradOp()
455 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in MaxPoolingGradGradOp()
464 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in MaxPoolingGradGradOp()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc37 OP_REQUIRES_OK(ctx, ctx->GetAttr("begin_mask", &begin_mask_)); in StridedSliceOp()
38 OP_REQUIRES_OK(ctx, ctx->GetAttr("end_mask", &end_mask_)); in StridedSliceOp()
39 OP_REQUIRES_OK(ctx, ctx->GetAttr("ellipsis_mask", &ellipsis_mask_)); in StridedSliceOp()
40 OP_REQUIRES_OK(ctx, ctx->GetAttr("new_axis_mask", &new_axis_mask_)); in StridedSliceOp()
41 OP_REQUIRES_OK(ctx, ctx->GetAttr("shrink_axis_mask", &shrink_axis_mask_)); in StridedSliceOp()
42 OP_REQUIRES_OK(ctx, ctx->GetAttr("Index", &index_type_)); in StridedSliceOp()
120 OP_REQUIRES_OK(ctx, ctx->GetAttr("begin_mask", &begin_mask_)); in StridedSliceGradOp()
121 OP_REQUIRES_OK(ctx, ctx->GetAttr("end_mask", &end_mask_)); in StridedSliceGradOp()
122 OP_REQUIRES_OK(ctx, ctx->GetAttr("ellipsis_mask", &ellipsis_mask_)); in StridedSliceGradOp()
123 OP_REQUIRES_OK(ctx, ctx->GetAttr("new_axis_mask", &new_axis_mask_)); in StridedSliceGradOp()
[all …]
Dmatmul_op.cc34 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_a", &transpose_a_)); in MatMulOp()
35 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MatMulOp()
37 OP_REQUIRES_OK(ctx, ctx->GetAttr("Ta", &a_type_)); in MatMulOp()
38 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tb", &b_type_)); in MatMulOp()
45 OP_REQUIRES_OK(ctx, ctx->GetAttr("a_is_sparse", &dummy_is_sparse)); in MatMulOp()
46 OP_REQUIRES_OK(ctx, ctx->GetAttr("b_is_sparse", &dummy_is_sparse)); in MatMulOp()
Dlrn_ops.cc29 OP_REQUIRES_OK(ctx, ctx->GetAttr("depth_radius", &depth_radius_)); in LRNOp()
32 OP_REQUIRES_OK(ctx, ctx->GetAttr("bias", &bias_)); in LRNOp()
33 OP_REQUIRES_OK(ctx, ctx->GetAttr("alpha", &alpha_)); in LRNOp()
34 OP_REQUIRES_OK(ctx, ctx->GetAttr("beta", &beta_)); in LRNOp()
81 OP_REQUIRES_OK(ctx, ctx->GetAttr("depth_radius", &depth_radius_)); in LRNGradOp()
84 OP_REQUIRES_OK(ctx, ctx->GetAttr("bias", &bias_)); in LRNGradOp()
85 OP_REQUIRES_OK(ctx, ctx->GetAttr("alpha", &alpha_)); in LRNGradOp()
86 OP_REQUIRES_OK(ctx, ctx->GetAttr("beta", &beta_)); in LRNGradOp()
Dfake_quantize_ops.cc90 OP_REQUIRES_OK(ctx, ctx->GetAttr("num_bits", &num_bits)); in FakeQuantWithMinMaxArgsOp()
96 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range)); in FakeQuantWithMinMaxArgsOp()
101 OP_REQUIRES_OK(ctx, ctx->GetAttr("min", &input_min)); in FakeQuantWithMinMaxArgsOp()
102 OP_REQUIRES_OK(ctx, ctx->GetAttr("max", &input_max)); in FakeQuantWithMinMaxArgsOp()
138 OP_REQUIRES_OK(ctx, ctx->GetAttr("num_bits", &num_bits)); in FakeQuantWithMinMaxArgsGradOp()
144 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range)); in FakeQuantWithMinMaxArgsGradOp()
149 OP_REQUIRES_OK(ctx, ctx->GetAttr("min", &input_min)); in FakeQuantWithMinMaxArgsGradOp()
150 OP_REQUIRES_OK(ctx, ctx->GetAttr("max", &input_max)); in FakeQuantWithMinMaxArgsGradOp()
188 OP_REQUIRES_OK(ctx, ctx->GetAttr("num_bits", &num_bits)); in FakeQuantWithMinMaxVarsOp()
194 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range)); in FakeQuantWithMinMaxVarsOp()
[all …]
Dxla_dequantize_op.cc32 OP_REQUIRES_OK(context, context->GetAttr("min_range", &min_range_)); in XlaDequantizeOp()
33 OP_REQUIRES_OK(context, context->GetAttr("max_range", &max_range_)); in XlaDequantizeOp()
34 OP_REQUIRES_OK(context, context->GetAttr("mode", &mode_)); in XlaDequantizeOp()
36 context->GetAttr("transpose_output", &transpose_output_)); in XlaDequantizeOp()
Dpooling_ops.cc49 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_int)); in PoolingOp()
54 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &stride_int)); in PoolingOp()
65 OP_REQUIRES_OK(ctx, ctx->GetAttr("padding", &padding)); in PoolingOp()
156 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in MaxPoolOp()
209 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in AvgPoolOp()
271 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_)); in MaxPoolGradOp()
272 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &stride_)); in MaxPoolGradOp()
274 OP_REQUIRES_OK(ctx, ctx->GetAttr("padding", &padding_)); in MaxPoolGradOp()
357 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format)); in MaxPool2DGradOp()
380 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_)); in AvgPoolGradOp()
[all …]
Dxla_svd_op.cc28 OP_REQUIRES_OK(ctx, ctx->GetAttr("max_iter", &max_iter_)); in XlaSvdOp()
29 OP_REQUIRES_OK(ctx, ctx->GetAttr("epsilon", &epsilon_)); in XlaSvdOp()
32 ctx->GetAttr("precision_config", &precision_config_attr)); in XlaSvdOp()
58 OP_REQUIRES_OK(ctx, ctx->GetAttr("compute_uv", &compute_uv_)); in SvdOp()
59 OP_REQUIRES_OK(ctx, ctx->GetAttr("full_matrices", &full_matrices_)); in SvdOp()
/external/tensorflow/tensorflow/contrib/image/kernels/
Dsingle_image_random_dot_stereograms_ops.cc80 OP_REQUIRES_OK(context, context->GetAttr("hidden_surface_removal", in SingleImageRandomDotStereogramsOp()
82 OP_REQUIRES_OK(context, context->GetAttr("convergence_dots_size", in SingleImageRandomDotStereogramsOp()
84 OP_REQUIRES_OK(context, context->GetAttr("dots_per_inch", &dots_per_inch)); in SingleImageRandomDotStereogramsOp()
86 context->GetAttr("eye_separation", &eye_separation)); in SingleImageRandomDotStereogramsOp()
87 OP_REQUIRES_OK(context, context->GetAttr("mu", &mu)); in SingleImageRandomDotStereogramsOp()
88 OP_REQUIRES_OK(context, context->GetAttr("normalize", &normalize)); in SingleImageRandomDotStereogramsOp()
89 OP_REQUIRES_OK(context, context->GetAttr("normalize_max", &normalize_max)); in SingleImageRandomDotStereogramsOp()
90 OP_REQUIRES_OK(context, context->GetAttr("normalize_min", &normalize_min)); in SingleImageRandomDotStereogramsOp()
91 OP_REQUIRES_OK(context, context->GetAttr("border_level", &border_level)); in SingleImageRandomDotStereogramsOp()
92 OP_REQUIRES_OK(context, context->GetAttr("number_colors", &number_colors)); in SingleImageRandomDotStereogramsOp()
[all …]
/external/tensorflow/tensorflow/core/ops/
Dtpu_embedding_ops.cc144 TF_RETURN_IF_ERROR(c->GetAttr("table_id", &table_id)); in RegisterPerTableLoadOpsForAlgorithmBody()
146 TF_RETURN_IF_ERROR(c->GetAttr("table_name", &table_name)); in RegisterPerTableLoadOpsForAlgorithmBody()
153 TF_RETURN_IF_ERROR(c->GetAttr("num_shards", &num_shards)); in RegisterPerTableLoadOpsForAlgorithmBody()
155 TF_RETURN_IF_ERROR(c->GetAttr("shard_id", &shard_id)); in RegisterPerTableLoadOpsForAlgorithmBody()
261 TF_RETURN_IF_ERROR(c->GetAttr("table_id", &table_id)); in RegisterPerTableRetrieveOpsForAlgorithmBody()
263 TF_RETURN_IF_ERROR(c->GetAttr("table_name", &table_name)); in RegisterPerTableRetrieveOpsForAlgorithmBody()
270 TF_RETURN_IF_ERROR(c->GetAttr("num_shards", &num_shards)); in RegisterPerTableRetrieveOpsForAlgorithmBody()
272 TF_RETURN_IF_ERROR(c->GetAttr("shard_id", &shard_id)); in RegisterPerTableRetrieveOpsForAlgorithmBody()
334 TF_RETURN_IF_ERROR(c->GetAttr("config", &config_string)); in __anon8eedc5730902()
374 TF_RETURN_IF_ERROR(c->GetAttr("NN", &nn)); in __anon8eedc5730b02()
[all …]
Dlist_ops.cc60 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonebbd57d80202()
78 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonebbd57d80302()
125 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonebbd57d80402()
167 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonebbd57d80502()
204 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonebbd57d80602()
227 TF_RETURN_IF_ERROR(c->GetAttr("num_elements", &expected_num_elements)); in __anonebbd57d80602()
244 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in TensorListConcatShapeInference()
284 TF_RETURN_IF_ERROR(c->GetAttr("element_shape", &raw_element_shape)); in __anonebbd57d80702()
316 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonebbd57d80902()
350 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonebbd57d80a02()
[all …]
Dboosted_trees_ops.cc60 TF_RETURN_IF_ERROR(c->GetAttr("max_splits", &max_splits)); in __anone3ce78ae0202()
61 TF_RETURN_IF_ERROR(c->GetAttr("num_features", &num_features)); in __anone3ce78ae0202()
156 TF_RETURN_IF_ERROR(c->GetAttr("max_splits", &max_splits)); in __anone3ce78ae0602()
157 TF_RETURN_IF_ERROR(c->GetAttr("num_buckets", &num_buckets)); in __anone3ce78ae0602()
158 TF_RETURN_IF_ERROR(c->GetAttr("num_features", &num_features)); in __anone3ce78ae0602()
196 c->GetAttr("num_bucketized_features", &num_bucketized_features)); in __anone3ce78ae0702()
205 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anone3ce78ae0702()
223 c->GetAttr("num_bucketized_features", &num_bucketized_features)); in __anone3ce78ae0802()
263 c->GetAttr("num_bucketized_features", &num_bucketized_features)); in __anone3ce78ae0a02()
276 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anone3ce78ae0a02()
[all …]
Dcheckpoint_ops.cc34 TF_RETURN_IF_ERROR(c->GetAttr("new_vocab_offset", &new_vocab_offset)); in __anond28703640102()
36 TF_RETURN_IF_ERROR(c->GetAttr("num_new_vocab", &num_new_vocab)); in __anond28703640102()
64 TF_RETURN_IF_ERROR(c->GetAttr("num_rows", &num_rows)); in __anond28703640202()
66 TF_RETURN_IF_ERROR(c->GetAttr("num_cols", &num_cols)); in __anond28703640202()
/external/tensorflow/tensorflow/contrib/cloud/kernels/
Dbigquery_reader_ops.cc38 TF_RETURN_IF_ERROR(context->GetAttr("project_id", project_id)); in GetTableAttrs()
39 TF_RETURN_IF_ERROR(context->GetAttr("dataset_id", dataset_id)); in GetTableAttrs()
40 TF_RETURN_IF_ERROR(context->GetAttr("table_id", table_id)); in GetTableAttrs()
41 TF_RETURN_IF_ERROR(context->GetAttr("timestamp_millis", timestamp_millis)); in GetTableAttrs()
42 TF_RETURN_IF_ERROR(context->GetAttr("columns", columns)); in GetTableAttrs()
43 TF_RETURN_IF_ERROR(context->GetAttr("test_end_point", test_end_point)); in GetTableAttrs()
176 TF_RETURN_IF_ERROR(context->GetAttr("num_partitions", &num_partitions_)); in InitializeNumberOfPartitions()

12345678910>>...16