Home
last modified time | relevance | path

Searched refs:OpInputList (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dtensor_utils.h31 const OpInputList& input_list);
35 OpInputList* features_list);
39 OpInputList* features_indices_list,
40 OpInputList* feature_values_list,
41 OpInputList* feature_shapes_list);
45 OpInputList* features_indices_list,
46 OpInputList* feature_values_list,
47 OpInputList* feature_shapes_list);
51 const OpInputList& dense_float_features_list,
52 const OpInputList& sparse_float_feature_shapes_list,
[all …]
Dtensor_utils.cc26 const OpInputList& input_list) { in OpInputListToTensorVec()
36 OpInputList* features_list) { in ReadDenseFloatFeatures()
47 OpInputList* features_indices_list, in ReadSparseFloatFeatures()
48 OpInputList* feature_values_list, in ReadSparseFloatFeatures()
49 OpInputList* feature_shapes_list) { in ReadSparseFloatFeatures()
67 OpInputList* features_indices_list, in ReadSparseIntFeatures()
68 OpInputList* feature_values_list, in ReadSparseIntFeatures()
69 OpInputList* feature_shapes_list) { in ReadSparseIntFeatures()
86 const OpInputList& dense_float_features_list, in InferBatchSize()
87 const OpInputList& sparse_float_feature_shapes_list, in InferBatchSize()
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dprediction_ops.cc191 OpInputList dense_float_features_list; in DoCompute()
196 OpInputList sparse_float_feature_indices_list; in DoCompute()
197 OpInputList sparse_float_feature_values_list; in DoCompute()
198 OpInputList sparse_float_feature_shapes_list; in DoCompute()
205 OpInputList sparse_int_feature_indices_list; in DoCompute()
206 OpInputList sparse_int_feature_values_list; in DoCompute()
207 OpInputList sparse_int_feature_shapes_list; in DoCompute()
403 OpInputList dense_float_features_list; in DoCompute()
408 OpInputList sparse_float_feature_indices_list; in DoCompute()
409 OpInputList sparse_float_feature_values_list; in DoCompute()
[all …]
Dquantile_ops.cc80 const OpInputList& buckets_list) { in GetBuckets()
88 const OpInputList* const indices_list) { in GetFeatureDimension()
99 const string& output_name, const OpInputList& values_list, in QuantizeFeatures()
100 const OpInputList& buckets_list, in QuantizeFeatures()
101 const OpInputList* const in QuantizeFeatures()
280 OpInputList resource_handle_list; in Compute()
283 OpInputList summary_list; in Compute()
360 OpInputList dense_float_features_list; in Compute()
365 OpInputList sparse_float_feature_indices_list; in Compute()
366 OpInputList sparse_float_feature_values_list; in Compute()
[all …]
Dstats_accumulator_ops.cc422 OpInputList resource_handle_list; in Compute()
425 OpInputList partition_ids_list; in Compute()
429 OpInputList feature_ids_list; in Compute()
432 OpInputList gradients_list; in Compute()
434 OpInputList hessians_list; in Compute()
487 OpInputList resource_handle_list; in Compute()
490 OpInputList partition_ids_list; in Compute()
494 OpInputList feature_ids_list; in Compute()
497 OpInputList gradients_list; in Compute()
499 OpInputList hessians_list; in Compute()
Dtraining_ops.cc365 OpInputList partition_ids_list; in Compute()
366 OpInputList gains_list; in Compute()
367 OpInputList splits_list; in Compute()
442 OpKernelContext* const context, const OpInputList& partition_ids_list, in FindBestSplitsPerPartitionNormal()
443 const OpInputList& gains_list, const OpInputList& splits_list, in FindBestSplitsPerPartitionNormal()
478 const OpInputList& gains_list, in FindBestSplitOblivious()
479 const OpInputList& splits_list, in FindBestSplitOblivious()
/external/tensorflow/tensorflow/core/kernels/
Dsparse_cross_op.cc301 OpInputList indices_list_in; in Compute()
303 OpInputList values_list_in; in Compute()
305 OpInputList shapes_list_in; in Compute()
307 OpInputList dense_list_in; in Compute()
352 const OpInputList& indices_list_in, in ValidateInput()
353 const OpInputList& values_list_in, in ValidateInput()
354 const OpInputList& shapes_list_in, in ValidateInput()
355 const OpInputList& dense_list_in) { in ValidateInput()
431 int64 CalculateBatchSize(const OpInputList& shapes_list_in, in CalculateBatchSize()
432 const OpInputList& dense_list_in) { in CalculateBatchSize()
[all …]
Dsdca_internal.cc96 OpInputList sparse_indices_inputs; in Initialize()
99 OpInputList sparse_weights_inputs; in Initialize()
102 OpInputList dense_weights_inputs; in Initialize()
130 [&](const OpInputList& weight_inputs, OpOutputList* const weight_outputs, in Initialize()
327 OpInputList sparse_example_indices_inputs; in Initialize()
330 OpInputList sparse_feature_indices_inputs; in Initialize()
333 OpInputList sparse_feature_values_inputs; in Initialize()
355 OpInputList dense_features_inputs; in Initialize()
389 const OpInputList& sparse_example_indices_inputs, in CreateSparseFeatureRepresentation()
390 const OpInputList& sparse_feature_indices_inputs, in CreateSparseFeatureRepresentation()
[all …]
Dquantized_concat_op.cc79 const OpInputList& input_mins, const OpInputList& input_maxes, in CalculateInputAndOutputRange()
118 int input_dims, const OpInputList& values, in CalculateConcatDims()
169 OpInputList values; in Compute()
172 OpInputList input_mins; in Compute()
178 OpInputList input_maxes; in Compute()
Ddynamic_stitch_op.cc76 OpInputList* indices_inputs, in CheckArgsAndAllocateResult()
77 OpInputList* data_inputs, int* first_dim_size, in CheckArgsAndAllocateResult()
164 OpInputList indices_inputs; in Compute()
165 OpInputList data_inputs; in Compute()
236 OpInputList indices_inputs; in Compute()
237 OpInputList data_inputs; in Compute()
Dsdca_internal.h354 const OpInputList& sparse_example_indices_inputs,
355 const OpInputList& sparse_feature_indices_inputs,
356 const OpInputList& sparse_feature_values_inputs,
365 const OpInputList& dense_features_inputs,
Dsparse_concat_op.cc42 OpInputList inds; in Compute()
52 OpInputList vals; in Compute()
64 OpInputList shapes; in Compute()
Dexample_parsing_ops.cc46 OpInputList dense_keys; in Compute()
47 OpInputList sparse_keys; in Compute()
48 OpInputList dense_defaults; in Compute()
179 OpInputList dense_defaults; in Compute()
276 OpInputList context_dense_defaults; in Compute()
436 OpInputList context_dense_keys; in Compute()
437 OpInputList context_sparse_keys; in Compute()
438 OpInputList context_dense_defaults; in Compute()
439 OpInputList feature_list_dense_keys; in Compute()
440 OpInputList feature_list_sparse_keys; in Compute()
Didentity_n_op.h28 OpInputList input; in Compute()
Dpartitioned_function_ops.cc117 OpInputList args; in ComputeAsync()
183 OpInputList args; in Instantiate()
/external/tensorflow/tensorflow/contrib/layers/kernels/
Dsparse_feature_cross_kernel.cc345 OpInputList indices_list_in; in Compute()
347 OpInputList values_list_in; in Compute()
349 OpInputList shapes_list_in; in Compute()
351 OpInputList dense_list_in; in Compute()
395 const OpInputList& indices_list_in, in ValidateInput()
396 const OpInputList& values_list_in, in ValidateInput()
397 const OpInputList& shapes_list_in, in ValidateInput()
398 const OpInputList& dense_list_in) { in ValidateInput()
474 int64 CalculateBatchSize(const OpInputList& shapes_list_in, in CalculateBatchSize()
475 const OpInputList& dense_list_in) { in CalculateBatchSize()
[all …]
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc66 OpInputList node_ids_list; in Compute()
67 OpInputList gains_list; in Compute()
68 OpInputList thresholds_list; in Compute()
69 OpInputList left_node_contribs; in Compute()
70 OpInputList right_node_contribs; in Compute()
197 OpKernelContext* const context, const OpInputList& node_ids_list, in FindBestSplitsPerNode()
198 const OpInputList& gains_list, in FindBestSplitsPerNode()
Dquantile_ops.cc82 const OpInputList& buckets_list) { in GetBuckets()
147 OpInputList float_features_list; in Compute()
234 OpInputList summaries_list; in Compute()
292 OpInputList bucket_boundaries_list; in Compute()
451 OpInputList float_features_list; in Compute()
454 OpInputList bucket_boundaries_list; in Compute()
Dprediction_ops.cc65 OpInputList bucketized_features_list; in Compute()
204 OpInputList bucketized_features_list; in Compute()
308 OpInputList bucketized_features_list; in Compute()
Dstats_ops.cc41 OpInputList stats_summary_list; in Compute()
231 OpInputList bucketized_features_list; in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op.cc137 OpInputList args;
139 OpInputList captured_inputs;
150 ComputeOptions(OpKernelContext* ctx, OpInputList args, in ComputeOptions()
151 OpInputList captured_inputs, in ComputeOptions()
267 OpInputList arguments; in SetupArgs()
269 OpInputList captured_inputs; in SetupArgs()
/external/tensorflow/tensorflow/lite/tools/accuracy/
Drun_tflite_model_op.cc31 Status ValidateInputsMatch(const OpInputList& input_tensors, in ValidateInputsMatch()
86 OpInputList input_tensors; in Compute()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h447 class OpInputList {
449 typedef OpArgIterator<OpInputList, const Tensor> Iterator;
450 OpInputList() : ctx_(nullptr), start_(0), stop_(0) {} in OpInputList() function
451 OpInputList(OpKernelContext* ctx, int start, int stop) in OpInputList() function
453 OpInputList& operator=(const OpInputList& other) = default;
728 Status input_list(StringPiece name, OpInputList* list);
1634 inline const Tensor& OpInputList::operator[](int i) const {
/external/tensorflow/tensorflow/core/kernels/lookup_tables/
Dop_kernel_templates.h173 OpInputList table_int64_args; in Compute()
303 OpInputList table_int64_args; in Compute()
349 OpInputList table_int64_args; in Compute()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h70 const xrt::XLATupleNode& tuple_node, const OpInputList& input_tensor_list, in ParseTupleNode()
122 const OpInputList& input_tensor_list, in ParseTupleTree()
256 OpInputList values; in Compute()
392 OpInputList arg_list; in Compute()

123