/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/ |
D | tensor_utils.h | 31 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 …]
|
D | tensor_utils.cc | 26 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/ |
D | prediction_ops.cc | 191 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 …]
|
D | quantile_ops.cc | 80 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 …]
|
D | stats_accumulator_ops.cc | 422 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()
|
D | training_ops.cc | 365 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/ |
D | sparse_cross_op.cc | 301 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 …]
|
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 | 46 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()
|
D | identity_n_op.h | 28 OpInputList input; in Compute()
|
D | partitioned_function_ops.cc | 117 OpInputList args; in ComputeAsync() 183 OpInputList args; in Instantiate()
|
/external/tensorflow/tensorflow/contrib/layers/kernels/ |
D | sparse_feature_cross_kernel.cc | 345 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/ |
D | training_ops.cc | 66 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()
|
D | quantile_ops.cc | 82 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()
|
D | prediction_ops.cc | 65 OpInputList bucketized_features_list; in Compute() 204 OpInputList bucketized_features_list; in Compute() 308 OpInputList bucketized_features_list; in Compute()
|
D | stats_ops.cc | 41 OpInputList stats_summary_list; in Compute() 231 OpInputList bucketized_features_list; in Compute()
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | map_defun_op.cc | 137 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/ |
D | run_tflite_model_op.cc | 31 Status ValidateInputsMatch(const OpInputList& input_tensors, in ValidateInputsMatch() 86 OpInputList input_tensors; in Compute()
|
/external/tensorflow/tensorflow/core/framework/ |
D | op_kernel.h | 447 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/ |
D | op_kernel_templates.h | 173 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/ |
D | xrt_state_ops.h | 70 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()
|