| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | training_ops.cc | 55 OpInputList node_ids_list; in Compute()
56 OpInputList gains_list; in Compute()
57 OpInputList thresholds_list; in Compute()
58 OpInputList left_node_contribs; in Compute()
59 OpInputList right_node_contribs; in Compute()
171 const OpInputList& node_ids_list, const OpInputList& gains_list, in FindBestSplitsPerNode()
172 const OpInputList& thresholds_list, in FindBestSplitsPerNode()
173 const OpInputList& left_node_contribs_list, in FindBestSplitsPerNode()
174 const OpInputList& right_node_contribs_list, in FindBestSplitsPerNode()
251 OpInputList node_ids_list; in Compute()
[all …]
|
| D | quantile_ops.cc | 83 const OpInputList& buckets_list) { in GetBuckets()
148 OpInputList float_features_list; in Compute()
294 OpInputList summaries_list; in Compute()
357 OpInputList bucket_boundaries_list; in Compute()
515 OpInputList float_features_list; in Compute()
518 OpInputList bucket_boundaries_list; in Compute()
|
| D | prediction_ops.cc | 40 const OpInputList bucketized_features_list, in ConvertVectorsToMatrices()
74 OpInputList bucketized_features_list; in Compute()
217 OpInputList bucketized_features_list; in Compute()
317 OpInputList bucketized_features_list; in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_cross_op.cc | 324 OpInputList ragged_values_list; in Compute()
325 OpInputList ragged_splits_list; in Compute()
326 OpInputList sparse_indices_list; in Compute()
327 OpInputList sparse_values_list; in Compute()
328 OpInputList sparse_shape_list; in Compute()
329 OpInputList dense_list; in Compute()
376 Status ValidateInput(const OpInputList& ragged_values_list, in ValidateInput()
377 const OpInputList& ragged_splits_list, in ValidateInput()
378 const OpInputList& sparse_indices_list, in ValidateInput()
379 const OpInputList& sparse_values_list, in ValidateInput()
[all …]
|
| D | sparse_cross_op.cc | 446 int64 CalculateBatchSize(const OpInputList& shapes_list_in, in CalculateBatchSize()
447 const OpInputList& dense_list_in) { in CalculateBatchSize()
460 Status ValidateInput(const OpInputList& indices_list_in, in ValidateInput()
461 const OpInputList& values_list_in, in ValidateInput()
462 const OpInputList& shapes_list_in, in ValidateInput()
463 const OpInputList& dense_list_in) { in ValidateInput()
551 const OpInputList& indices_list_in, int64 batch_size, in ExtractFeatureData()
592 GenerateColumnsFromInput(const OpInputList& indices_list_in, in GenerateColumnsFromInput()
593 const OpInputList& values_list_in, in GenerateColumnsFromInput()
594 const OpInputList& shapes_list_in, in GenerateColumnsFromInput()
[all …]
|
| D | sdca_internal.cc | 96 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 …]
|
| D | quantized_concat_op.cc | 79 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()
|
| D | dynamic_stitch_op.cc | 76 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()
|
| D | sdca_internal.h | 354 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,
|
| D | sparse_concat_op.cc | 42 OpInputList inds; in Compute()
52 OpInputList vals; in Compute()
64 OpInputList shapes; in Compute()
|
| D | example_parsing_ops.cc | 61 OpInputList dense_defaults; in Compute()
116 OpInputList key_list; in GetInputListKeys()
127 const OpInputList& dense_defaults, in CheckInputShapes()
212 const OpInputList& dense_defaults) const { in MakeConfig()
306 OpInputList dense_defaults; in Compute()
409 OpInputList context_dense_defaults; in Compute()
489 const OpInputList& context_dense_defaults, in CheckInputShapes()
571 const OpInputList& context_dense_defaults) const { in MakeContextConfig()
787 OpInputList context_dense_keys; in Compute()
788 OpInputList context_sparse_keys; in Compute()
[all …]
|
| D | identity_n_op.h | 28 OpInputList input; in Compute()
|
| D | string_join_op.cc | 38 OpInputList input_list; in Compute()
|
| D | partitioned_function_ops.cc | 123 OpInputList args; in ComputeAsync()
208 OpInputList args; in Instantiate()
|
| D | ragged_gather_op.cc | 61 OpInputList params_nested_splits_in; in Compute()
116 const Tensor& indices_in, const OpInputList& params_nested_splits_in, in MakeSplits()
|
| D | queue_op.cc | 102 OpInputList components; in ComputeAsync()
143 OpInputList components; in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 47 OpInputList args;
49 OpInputList captured_inputs;
60 ComputeOptions(OpKernelContext* ctx, OpInputList args, in ComputeOptions()
61 OpInputList captured_inputs, in ComputeOptions()
257 OpInputList arguments; in SetupArgs()
259 OpInputList captured_inputs; in SetupArgs()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 99 const OpInputList& input_mins, const OpInputList& input_maxes, in CalculateInputAndOutputRange()
134 const OpInputList& input_mins, const OpInputList& input_maxes, in Compute()
457 OpInputList input_tensors; in Compute()
539 OpInputList input_mins, input_maxes; in Compute()
817 void CallEigenVersion(OpKernelContext* context, const OpInputList& values, in CallEigenVersion()
818 const OpInputList& input_mins, in CallEigenVersion()
819 const OpInputList& input_maxes, in CallEigenVersion()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_op_util.cc | 87 const OpInputList& guaranteed_constants) { in GuaranteedConstFingerprint()
106 uint64 mlir_module_fingerprint, const OpInputList& guaranteed_constants, in CreateCompilationCacheKey()
|
| D | tpu_op_util.h | 37 uint64 mlir_module_fingerprint, const OpInputList& guaranteed_constants,
|
| D | tpu_compile_op_common.h | 123 const OpInputList& guaranteed_constants,
134 const OpInputList& guaranteed_constants, in LookupPersistentCompilationCacheAndFillCaches()
|
| D | tpu_util.h | 55 Status DynamicShapesToTensorShapes(const OpInputList& dynamic_shapes,
|
| D | tpu_util.cc | 79 Status DynamicShapesToTensorShapes(const OpInputList& dynamic_shapes, in DynamicShapesToTensorShapes()
|
| D | tpu_compile_op_support.h | 55 const OpInputList* const>;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 443 class OpInputList {
445 typedef OpArgIterator<OpInputList, const Tensor> Iterator;
446 OpInputList() : ctx_(nullptr), start_(0), stop_(0) {} in OpInputList() function
447 OpInputList(OpKernelContext* ctx, int start, int stop) in OpInputList() function
449 OpInputList& operator=(const OpInputList& other) = default;
759 Status input_list(StringPiece name, OpInputList* list);
1674 inline const Tensor& OpInputList::operator[](int i) const {
|