Home
last modified time | relevance | path

Searched refs:OpOutputList (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dctc_decoder_ops.cc58 Tensor** log_prob, OpOutputList* decoded_indices, in ValidateInputsGenerateOutputs()
59 OpOutputList* decoded_values, OpOutputList* decoded_shape) const { in ValidateInputsGenerateOutputs()
114 OpOutputList* decoded_indices, OpOutputList* decoded_values, in StoreAllDecodedSequences()
115 OpOutputList* decoded_shape) const { in StoreAllDecodedSequences()
182 OpOutputList decoded_indices; in Compute()
183 OpOutputList decoded_values; in Compute()
184 OpOutputList decoded_shape; in Compute()
261 OpOutputList decoded_indices; in Compute()
262 OpOutputList decoded_values; in Compute()
263 OpOutputList decoded_shape; in Compute()
Dexample_parsing_ops.cc146 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()
407 OpOutputList context_sparse_indices; in Compute()
408 OpOutputList context_sparse_values; in Compute()
[all …]
Ddynamic_partition_op_gpu.cu.cc212 OpOutputList* Tout, DoneCallback done) { in AllocateOutputs()
253 OpOutputList outputs; in ComputeAsync()
305 OpOutputList outputs; in ComputeAsync()
439 OpOutputList& outs) { in GatherSlices()
Ddynamic_partition_op.cc44 OpOutputList* Tout) { in ValidateAndAllocateOutputs()
93 OpOutputList outputs; in Compute()
Dqueue_ops.cc202 OpOutputList output_components; in ComputeAsync()
267 OpOutputList output_components; in ComputeAsync()
353 OpOutputList output_components; in ComputeAsync()
Didentity_n_op.h29 OpOutputList output; in Compute()
Dsdca_internal.cc100 OpOutputList sparse_weights_outputs; in Initialize()
104 OpOutputList dense_weights_outputs; in Initialize()
124 [&](const OpInputList& weight_inputs, OpOutputList* const weight_outputs, in Initialize()
Dcritical_section.cc202 OpOutputList output; in DoneAndDelete()
Ddecode_csv_op.cc60 OpOutputList output; in Compute()
Dbarrier_ops.cc614 OpOutputList values_output; in ComputeAsync()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h432 class OpOutputList {
434 typedef OpArgIterator<OpOutputList, const Tensor*> Iterator;
435 OpOutputList() : ctx_(nullptr), start_(0), stop_(0) {} in OpOutputList() function
436 OpOutputList(OpKernelContext* ctx, int start, int stop) in OpOutputList() function
438 OpOutputList& operator=(const OpOutputList& other) = default;
749 Status output_list(StringPiece name, OpOutputList* list);
1454 inline Tensor* OpOutputList::operator[](int i) {
1460 inline bool OpOutputList::required(int i) const { in required()
1466 inline DataType OpOutputList::expected_output_dtype(int i) const { in expected_output_dtype()
1472 inline Status OpOutputList::allocate(int i, const TensorShape& shape, in allocate()
[all …]
Dop_kernel_test.cc621 TEST_F(OpKernelBuilderTest, OpOutputList) { in TEST_F() argument
640 OpOutputList out_list; in TEST_F()
Dop_kernel.cc582 Status OpKernelContext::output_list(StringPiece name, OpOutputList* list) { in output_list()
585 *list = OpOutputList(this, start, stop); in output_list()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dquantile_ops.cc107 OpOutputList output_list; in QuantizeFeatures()
196 OpOutputList* output_list) { in CopyBoundaries()
372 OpOutputList sparse_summaries_output_list; in Compute()
376 OpOutputList dense_summaries_output_list; in Compute()
383 OpOutputList* output_list) { in Compute()
687 OpOutputList are_buckets_ready_list; in Compute()
690 OpOutputList buckets_list; in Compute()
785 OpOutputList sparse_buckets_output_list; in Compute()
788 OpOutputList dense_buckets_output_list; in Compute()