Home
last modified time | relevance | path

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

12345678910>>...18

/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.h166 TF_RETURN_IF_ERROR(ctx->GetAttr("sparse_types", &sparse_types));
167 TF_RETURN_IF_ERROR(ctx->GetAttr("Tdense", &dense_types));
168 TF_RETURN_IF_ERROR(ctx->GetAttr("dense_shapes", &dense_shapes));
173 TF_RETURN_IF_ERROR(ctx->GetAttr("Nsparse", &num_sparse));
174 TF_RETURN_IF_ERROR(ctx->GetAttr("Ndense", &num_dense));
178 ctx->GetAttr("ragged_value_types", &ragged_value_types));
179 TF_RETURN_IF_ERROR(ctx->GetAttr("num_sparse", &num_sparse));
181 ctx->GetAttr("ragged_split_types", &ragged_split_types));
210 TF_RETURN_IF_ERROR(ctx->GetAttr("sparse_keys", &sparse_keys)); in Init()
211 TF_RETURN_IF_ERROR(ctx->GetAttr("sparse_types", &sparse_types)); 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 __anon74d2595d0102()
72 TF_RETURN_IF_ERROR(ctx->GetAttr("window_size", &window_size)); in __anon74d2595d0102()
75 TF_RETURN_IF_ERROR(ctx->GetAttr("window_step", &window_step)); in __anon74d2595d0102()
79 TF_RETURN_IF_ERROR(ctx->GetAttr("num_channels", &num_channels)); in __anon74d2595d0102()
81 TF_RETURN_IF_ERROR(ctx->GetAttr("left_context", &left_context)); in __anon74d2595d0102()
83 TF_RETURN_IF_ERROR(ctx->GetAttr("right_context", &right_context)); in __anon74d2595d0102()
85 TF_RETURN_IF_ERROR(ctx->GetAttr("frame_stride", &frame_stride)); in __anon74d2595d0102()
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/
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()
96 OP_REQUIRES_OK(context, context->GetAttr("shape", &shape_)); in ScopedAllocatorConcatOp()
97 OP_REQUIRES_OK(context, context->GetAttr("T", &dtype_)); in ScopedAllocatorConcatOp()
98 OP_REQUIRES_OK(context, context->GetAttr("reshape", &reshape_)); in ScopedAllocatorConcatOp()
100 OP_REQUIRES_OK(context, context->GetAttr("sa_name", &name_)); in ScopedAllocatorConcatOp()
101 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 …]
Dcollective_ops.cc143 OP_REQUIRES_OK(c, c->GetAttr("group_size", &col_params_->group.group_size)); in CollectiveGatherOpKernel()
148 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_->group.group_key)); in CollectiveGatherOpKernel()
150 c, c->GetAttr("instance_key", &col_params_->instance.instance_key)); in CollectiveGatherOpKernel()
151 OP_REQUIRES_OK(c, c->GetAttr("T", &col_params_->instance.data_type)); in CollectiveGatherOpKernel()
153 c, c->GetAttr("communication_hint", in CollectiveGatherOpKernel()
156 c, c->GetAttr("timeout_seconds", in CollectiveGatherOpKernel()
214 OP_REQUIRES_OK(c, c->GetAttr("group_size", &col_params_->group.group_size)); in CollectiveReduceOpKernel()
219 OP_REQUIRES_OK(c, c->GetAttr("group_key", &col_params_->group.group_key)); in CollectiveReduceOpKernel()
221 c, c->GetAttr("instance_key", &col_params_->instance.instance_key)); in CollectiveReduceOpKernel()
223 c, c->GetAttr("subdiv_offsets", in CollectiveReduceOpKernel()
[all …]
Dfake_quant_ops.cc70 OP_REQUIRES_OK(context, context->GetAttr("min", &min_)); in FakeQuantWithMinMaxArgsOp()
71 OP_REQUIRES_OK(context, context->GetAttr("max", &max_)); in FakeQuantWithMinMaxArgsOp()
76 OP_REQUIRES_OK(context, context->GetAttr("num_bits", &num_bits)); in FakeQuantWithMinMaxArgsOp()
81 OP_REQUIRES_OK(context, context->GetAttr("narrow_range", &narrow_range)); in FakeQuantWithMinMaxArgsOp()
111 OP_REQUIRES_OK(context, context->GetAttr("min", &min_)); in FakeQuantWithMinMaxArgsGradientOp()
112 OP_REQUIRES_OK(context, context->GetAttr("max", &max_)); in FakeQuantWithMinMaxArgsGradientOp()
117 OP_REQUIRES_OK(context, context->GetAttr("num_bits", &num_bits)); in FakeQuantWithMinMaxArgsGradientOp()
122 OP_REQUIRES_OK(context, context->GetAttr("narrow_range", &narrow_range)); in FakeQuantWithMinMaxArgsGradientOp()
190 OP_REQUIRES_OK(context, context->GetAttr("num_bits", &num_bits)); in FakeQuantWithMinMaxVarsOp()
195 OP_REQUIRES_OK(context, context->GetAttr("narrow_range", &narrow_range)); in FakeQuantWithMinMaxVarsOp()
[all …]
Dfused_eigen_output_kernels.cc32 TF_RETURN_IF_ERROR(context->GetAttr("fused_ops", &fused_ops)); in InitializeFusedComputation()
39 TF_RETURN_IF_ERROR(context->GetAttr("num_args", &num_args)); in InitializeFusedComputation()
71 TF_RETURN_IF_ERROR(context->GetAttr( in InitializeFusedComputation()
89 context->GetAttr("epsilon", &fused_computation_args->epsilon)); in InitializeFusedComputation()
92 TF_RETURN_IF_ERROR(context->GetAttr( in InitializeFusedComputation()
Davgpooling_op.cc60 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in AvgPoolingOp()
68 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in AvgPoolingOp()
72 OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_)); in AvgPoolingOp()
76 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in AvgPoolingOp()
138 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in AvgPoolingOp()
141 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in AvgPoolingOp()
145 OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_)); in AvgPoolingOp()
149 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in AvgPoolingOp()
254 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in AvgPoolingGradOp()
262 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in AvgPoolingGradOp()
[all …]
Dquantize_and_dequantize_op.cc47 OP_REQUIRES_OK(ctx, ctx->GetAttr("signed_input", &signed_input_)); in QuantizeAndDequantizeV2Op()
48 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in QuantizeAndDequantizeV2Op()
49 OP_REQUIRES_OK(ctx, ctx->GetAttr("num_bits", &num_bits_)); in QuantizeAndDequantizeV2Op()
53 OP_REQUIRES_OK(ctx, ctx->GetAttr("range_given", &range_given_)); in QuantizeAndDequantizeV2Op()
56 OP_REQUIRES_OK(ctx, ctx->GetAttr("round_mode", &round_mode_string)); in QuantizeAndDequantizeV2Op()
69 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range_)); in QuantizeAndDequantizeV2Op()
148 OP_REQUIRES_OK(ctx, ctx->GetAttr("axis", &axis_)); in QuantizeAndDequantizeV4GradientOp()
219 OP_REQUIRES_OK(ctx, ctx->GetAttr("signed_input", &signed_input_)); in QuantizeAndDequantizeV3Op()
220 OP_REQUIRES_OK(ctx, ctx->GetAttr("range_given", &range_given_)); in QuantizeAndDequantizeV3Op()
221 OP_REQUIRES_OK(ctx, ctx->GetAttr("narrow_range", &narrow_range_)); in QuantizeAndDequantizeV3Op()
[all …]
Dsendrecv_ops.cc60 OP_REQUIRES_OK(ctx, ctx->GetAttr("send_device", &send_device)); in SendOp()
62 OP_REQUIRES_OK(ctx, ctx->GetAttr("recv_device", &recv_device)); in SendOp()
65 ctx, ctx->GetAttr("send_device_incarnation", in SendOp()
68 OP_REQUIRES_OK(ctx, ctx->GetAttr("tensor_name", &tensor_name)); in SendOp()
75 if (!ctx->GetAttr("_hostmem_sendrecv", &hostmem_sendrecv_).ok()) { in SendOp()
139 OP_REQUIRES_OK(ctx, ctx->GetAttr("send_device", &send_device)); in RecvOp()
141 OP_REQUIRES_OK(ctx, ctx->GetAttr("recv_device", &recv_device)); in RecvOp()
144 ctx, ctx->GetAttr("send_device_incarnation", in RecvOp()
147 OP_REQUIRES_OK(ctx, ctx->GetAttr("tensor_name", &tensor_name)); in RecvOp()
154 if (!ctx->GetAttr("_hostmem_sendrecv", &hostmem_sendrecv_).ok()) { in RecvOp()
Dstrided_slice_op.cc85 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceOp()
86 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_mask)); in StridedSliceOp()
87 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceOp()
88 OP_REQUIRES_OK(context, context->GetAttr("new_axis_mask", &new_axis_mask)); in StridedSliceOp()
90 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceOp()
193 OP_REQUIRES_OK(context, context->GetAttr("begin_mask", &begin_mask)); in StridedSliceGradOp()
194 OP_REQUIRES_OK(context, context->GetAttr("end_mask", &end_mask)); in StridedSliceGradOp()
195 OP_REQUIRES_OK(context, context->GetAttr("ellipsis_mask", &ellipsis_mask)); in StridedSliceGradOp()
196 OP_REQUIRES_OK(context, context->GetAttr("new_axis_mask", &new_axis_mask)); in StridedSliceGradOp()
198 context->GetAttr("shrink_axis_mask", &shrink_axis_mask)); in StridedSliceGradOp()
[all …]
Dbatch_kernels.cc190 OP_REQUIRES_OK(c, c->GetAttr("container", &container_)); in BatchFunctionKernel()
191 OP_REQUIRES_OK(c, c->GetAttr("shared_name", &shared_name_)); in BatchFunctionKernel()
192 OP_REQUIRES_OK(c, c->GetAttr("batching_queue", &batcher_queue_)); in BatchFunctionKernel()
193 OP_REQUIRES_OK(c, c->GetAttr("num_batch_threads", &num_batch_threads_)); in BatchFunctionKernel()
194 OP_REQUIRES_OK(c, c->GetAttr("max_batch_size", &max_batch_size_)); in BatchFunctionKernel()
196 c->GetAttr("batch_timeout_micros", &batch_timeout_micros_)); in BatchFunctionKernel()
198 c->GetAttr("max_enqueued_batches", &max_enqueued_batches_)); in BatchFunctionKernel()
199 OP_REQUIRES_OK(c, c->GetAttr("allowed_batch_sizes", &allowed_batch_sizes_)); in BatchFunctionKernel()
201 OP_REQUIRES_OK(c, c->GetAttr("f", &func_)); in BatchFunctionKernel()
233 OP_REQUIRES_OK(c, c->GetAttr("enable_large_batch_splitting", in BatchFunctionKernel()
[all …]
Dmaxpooling_op.cc225 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in MaxPoolingGradOp()
235 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in MaxPoolingGradOp()
239 OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_)); in MaxPoolingGradOp()
252 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in MaxPoolingGradOp()
256 context, context->GetAttr("explicit_paddings", &explicit_paddings_)); in MaxPoolingGradOp()
351 OP_REQUIRES_OK(context, context->GetAttr("data_format", &data_format)); in MaxPoolingGradOp()
355 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in MaxPoolingGradOp()
359 OP_REQUIRES_OK(context, context->GetAttr("strides", &stride_)); in MaxPoolingGradOp()
369 OP_REQUIRES_OK(context, context->GetAttr("padding", &padding_)); in MaxPoolingGradOp()
372 context, context->GetAttr("explicit_paddings", &explicit_paddings_)); in MaxPoolingGradOp()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatmul_op.cc37 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_a", &transpose_a_)); in MatMulOp()
38 OP_REQUIRES_OK(ctx, ctx->GetAttr("transpose_b", &transpose_b_)); in MatMulOp()
40 OP_REQUIRES_OK(ctx, ctx->GetAttr("Ta", &a_type_)); in MatMulOp()
41 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tb", &b_type_)); in MatMulOp()
48 OP_REQUIRES_OK(ctx, ctx->GetAttr("a_is_sparse", &dummy_is_sparse)); in MatMulOp()
49 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()
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()
Dgather_scatter_ops.cc29 OP_REQUIRES_OK(context, context->GetAttr("dimension_numbers", &dnums_attr)); in GatherOp()
34 context, context->GetAttr("indices_are_sorted", &indices_are_sorted_)); in GatherOp()
59 context, context->GetAttr("update_computation", &update_computation_)); in ScatterOp()
61 OP_REQUIRES_OK(context, context->GetAttr("dimension_numbers", &dnums_attr)); in ScatterOp()
66 context, context->GetAttr("indices_are_sorted", &indices_are_sorted_)); in ScatterOp()
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()
Dpooling_ops.cc51 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_int)); in PoolingOp()
56 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &stride_int)); in PoolingOp()
67 OP_REQUIRES_OK(ctx, ctx->GetAttr("padding", &padding)); in PoolingOp()
158 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in MaxPoolOp()
217 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format_str)); in AvgPoolOp()
278 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_)); in MaxPoolGradOp()
279 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &stride_)); in MaxPoolGradOp()
281 OP_REQUIRES_OK(ctx, ctx->GetAttr("padding", &padding_)); in MaxPoolGradOp()
378 OP_REQUIRES_OK(ctx, ctx->GetAttr("data_format", &data_format)); in MaxPool2DGradOp()
401 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_)); in AvgPoolGradOp()
[all …]
/external/tensorflow/tensorflow/core/kernels/image/
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()
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc74 OP_REQUIRES_OK(ctx, ctx->GetAttr("cost_estimate_ns", &cost_estimate_)); in HostComputeOp()
75 OP_REQUIRES_OK(ctx, ctx->GetAttr("key", &key_)); in HostComputeOp()
76 OP_REQUIRES_OK(ctx, ctx->GetAttr("tpu_core", &tpu_core_)); in HostComputeOp()
77 OP_REQUIRES_OK(ctx, ctx->GetAttr("Tinputs", &input_dtypes_)); in HostComputeOp()
82 OP_REQUIRES_OK(ctx, ctx->GetAttr("Toutputs", &output_dtypes_)); in HostComputeOp()
87 OP_REQUIRES_OK(ctx, ctx->GetAttr("ancestors", &ancestors_)); in HostComputeOp()
90 ctx, ctx->GetAttr("shape_inference_graph", &shape_inference_graph)); in HostComputeOp()
92 OP_REQUIRES_OK(ctx, ctx->GetAttr("shapes", &static_output_shapes_)); in HostComputeOp()
119 ctx, ctx->GetAttr(kXlaTokenInputNodesAttrName, &token_input_nodes_)); in HostComputeOp()
123 OP_REQUIRES_OK(ctx, ctx->GetAttr(kXlaOriginalOutsideCompilationNodeName, in HostComputeOp()
[all …]
/external/tensorflow/tensorflow/core/ops/
Dragged_array_ops.cc70 c->GetAttr("ragged_values_types", &ragged_values_types)); in __anon9deb1dc20102()
72 c->GetAttr("ragged_splits_types", &ragged_splits_types)); in __anon9deb1dc20102()
73 TF_RETURN_IF_ERROR(c->GetAttr("dense_types", &dense_types)); in __anon9deb1dc20102()
82 TF_RETURN_IF_ERROR(c->GetAttr("Nsparse", &num_sparse)); in __anon9deb1dc20102()
125 c->GetAttr<int64>("PARAMS_RAGGED_RANK", &PARAMS_RAGGED_RANK)); in RaggedGatherShapeFn()
126 TF_RETURN_IF_ERROR(c->GetAttr<int>("OUTPUT_RAGGED_RANK", &num_splits)); in RaggedGatherShapeFn()
Dlist_ops.cc66 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonabedecb60202()
84 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonabedecb60302()
131 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonabedecb60402()
173 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonabedecb60502()
210 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonabedecb60602()
238 TF_RETURN_IF_ERROR(c->GetAttr("num_elements", &expected_num_elements)); in __anonabedecb60602()
255 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in TensorListConcatShapeInference()
295 TF_RETURN_IF_ERROR(c->GetAttr("element_shape", &raw_element_shape)); in __anonabedecb60702()
327 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonabedecb60902()
361 TF_RETURN_IF_ERROR(c->GetAttr("element_dtype", &element_dtype)); in __anonabedecb60a02()
[all …]
Dcheckpoint_ops.cc34 TF_RETURN_IF_ERROR(c->GetAttr("new_vocab_offset", &new_vocab_offset)); in __anon62d746c20102()
36 TF_RETURN_IF_ERROR(c->GetAttr("num_new_vocab", &num_new_vocab)); in __anon62d746c20102()
64 TF_RETURN_IF_ERROR(c->GetAttr("num_rows", &num_rows)); in __anon62d746c20202()
66 TF_RETURN_IF_ERROR(c->GetAttr("num_cols", &num_cols)); in __anon62d746c20202()
Dboosted_trees_ops.cc61 TF_RETURN_IF_ERROR(c->GetAttr("max_splits", &max_splits)); in __anon770f674c0202()
62 TF_RETURN_IF_ERROR(c->GetAttr("num_features", &num_features)); in __anon770f674c0202()
136 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anon770f674c0302()
166 TF_RETURN_IF_ERROR(c->GetAttr("num_features", &num_features)); in __anon770f674c0402()
168 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anon770f674c0402()
254 TF_RETURN_IF_ERROR(c->GetAttr("logits_dimension", &logits_dimension)); in __anon770f674c0502()
318 TF_RETURN_IF_ERROR(c->GetAttr("max_splits", &max_splits)); in __anon770f674c0902()
319 TF_RETURN_IF_ERROR(c->GetAttr("num_buckets", &num_buckets)); in __anon770f674c0902()
320 TF_RETURN_IF_ERROR(c->GetAttr("num_features", &num_features)); in __anon770f674c0902()
360 TF_RETURN_IF_ERROR(c->GetAttr("max_splits", &max_splits)); in __anon770f674c0a02()
[all …]

12345678910>>...18