| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 59 Tensor** log_prob, OpOutputList* decoded_indices, in ValidateInputsGenerateOutputs()
60 OpOutputList* decoded_values, OpOutputList* decoded_shape) const { in ValidateInputsGenerateOutputs()
115 OpOutputList* decoded_indices, OpOutputList* decoded_values, in StoreAllDecodedSequences()
116 OpOutputList* decoded_shape) const { in StoreAllDecodedSequences()
183 OpOutputList decoded_indices; in Compute()
184 OpOutputList decoded_values; in Compute()
185 OpOutputList decoded_shape; in Compute()
271 OpOutputList decoded_indices; in Compute()
272 OpOutputList decoded_values; in Compute()
273 OpOutputList decoded_shape; in Compute()
|
| D | example_parsing_ops.cc | 146 OpOutputList dense_values; in Compute()
147 OpOutputList sparse_indices; in Compute()
148 OpOutputList sparse_values; in Compute()
149 OpOutputList sparse_shapes; in Compute()
242 OpOutputList dense_values; in Compute()
243 OpOutputList sparse_indices; in Compute()
244 OpOutputList sparse_values; in Compute()
245 OpOutputList sparse_shapes; in Compute()
368 OpOutputList context_sparse_indices; in Compute()
369 OpOutputList context_sparse_values; in Compute()
[all …]
|
| D | queue_op.cc | 183 OpOutputList output_components; in ComputeAsync()
238 OpOutputList output_components; in ComputeAsync()
311 OpOutputList output_components; in ComputeAsync()
|
| D | dynamic_partition_op.cc | 44 OpOutputList* Tout) { in ValidateAndAllocateOutputs()
93 OpOutputList outputs; in Compute()
|
| D | dynamic_partition_op_gpu.cu.cc | 213 OpOutputList* Tout, DoneCallback done) { in AllocateOutputs()
254 OpOutputList outputs; in ComputeAsync()
306 OpOutputList outputs; in ComputeAsync()
440 OpOutputList& outs) { in GatherSlices()
|
| D | identity_n_op.h | 29 OpOutputList output; in Compute()
|
| D | sdca_internal.cc | 106 OpOutputList sparse_weights_outputs; in Initialize()
110 OpOutputList dense_weights_outputs; in Initialize()
130 [&](const OpInputList& weight_inputs, OpOutputList* const weight_outputs, in Initialize()
|
| D | scoped_allocator_ops_test.cc | 180 OpOutputList output_list; in ExecOp()
302 OpOutputList output_list; in ExecOp()
|
| D | ragged_gather_op.cc | 215 OpOutputList splits_out; in WriteSplits()
|
| D | decode_csv_op.cc | 76 OpOutputList output; in Compute()
|
| D | barrier_ops.cc | 613 OpOutputList values_output; in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 66 OpOutputList output_node_ids_list; in Compute()
69 OpOutputList output_gains_list; in Compute()
72 OpOutputList output_thresholds_list; in Compute()
75 OpOutputList output_left_node_contribs_list; in Compute()
79 OpOutputList output_right_node_contribs_list; in Compute()
|
| D | quantile_ops.cc | 167 OpOutputList summaries_output_list; in Compute()
403 OpOutputList bucket_boundaries_list; in Compute()
462 OpOutputList buckets_list; in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 488 class OpOutputList {
490 typedef OpArgIterator<OpOutputList, const Tensor*> Iterator;
491 OpOutputList() : ctx_(nullptr), start_(0), stop_(0) {} in OpOutputList() function
492 OpOutputList(OpKernelContext* ctx, int start, int stop) in OpOutputList() function
494 OpOutputList& operator=(const OpOutputList& other) = default;
885 Status output_list(StringPiece name, OpOutputList* list);
1652 inline Tensor* OpOutputList::operator[](int i) {
1658 inline bool OpOutputList::required(int i) const { in required()
1664 inline DataType OpOutputList::expected_output_dtype(int i) const { in expected_output_dtype()
1670 inline Status OpOutputList::allocate(int i, const TensorShape& shape, in allocate()
[all …]
|
| D | op_kernel_test.cc | 655 TEST_F(OpKernelBuilderTest, OpOutputList) { in TEST_F() argument
674 OpOutputList out_list; in TEST_F()
|
| D | op_kernel.cc | 650 Status OpKernelContext::output_list(StringPiece name, OpOutputList* list) { in output_list()
653 *list = OpOutputList(this, start, stop); in output_list()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 107 OpOutputList output_list; in QuantizeFeatures()
198 OpOutputList* output_list) { in CopyBoundaries()
380 OpOutputList sparse_summaries_output_list; in Compute()
384 OpOutputList dense_summaries_output_list; in Compute()
391 OpOutputList* output_list) { in Compute()
695 OpOutputList are_buckets_ready_list; in Compute()
698 OpOutputList buckets_list; in Compute()
794 OpOutputList sparse_buckets_output_list; in Compute()
797 OpOutputList dense_buckets_output_list; in Compute()
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ |
| D | run_tflite_model_op.cc | 90 OpOutputList output_tensors; in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 145 OpOutputList output GUARDED_BY(mu);
|