Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dctc_decoder_ops.cc61 Tensor** log_prob, OpOutputList* decoded_indices, in ValidateInputsGenerateOutputs()
62 OpOutputList* decoded_values, OpOutputList* decoded_shape) const { in ValidateInputsGenerateOutputs()
117 OpOutputList* decoded_indices, OpOutputList* decoded_values, in StoreAllDecodedSequences()
118 OpOutputList* decoded_shape) const { in StoreAllDecodedSequences()
193 OpOutputList decoded_indices; in Compute()
194 OpOutputList decoded_values; in Compute()
195 OpOutputList decoded_shape; in Compute()
291 OpOutputList decoded_indices; in Compute()
292 OpOutputList decoded_values; in Compute()
293 OpOutputList decoded_shape; in Compute()
Dexample_parsing_ops.cc257 OpOutputList dense_values; in WriteOutput()
258 OpOutputList sparse_indices; in WriteOutput()
259 OpOutputList sparse_values; in WriteOutput()
260 OpOutputList sparse_shapes; in WriteOutput()
274 OpOutputList ragged_values; in WriteOutput()
275 OpOutputList ragged_splits; in WriteOutput()
369 OpOutputList dense_values; in Compute()
370 OpOutputList sparse_indices; in Compute()
371 OpOutputList sparse_values; in Compute()
372 OpOutputList sparse_shapes; in Compute()
[all …]
Dqueue_op.cc183 OpOutputList output_components; in ComputeAsync()
238 OpOutputList output_components; in ComputeAsync()
311 OpOutputList output_components; in ComputeAsync()
Didentity_n_op.h29 OpOutputList output; in Compute()
Ddynamic_partition_op.cc44 OpOutputList* Tout) { in ValidateAndAllocateOutputs()
93 OpOutputList outputs; in Compute()
Ddynamic_partition_op_gpu.cu.cc220 OpOutputList* Tout, DoneCallback done) { in AllocateOutputs()
261 OpOutputList outputs; in ComputeAsync()
316 OpOutputList outputs; in ComputeAsync()
456 OpOutputList& outs) { in GatherSlices()
Dsdca_internal.cc106 OpOutputList sparse_weights_outputs; in Initialize()
110 OpOutputList dense_weights_outputs; in Initialize()
130 [&](const OpInputList& weight_inputs, OpOutputList* const weight_outputs, in Initialize()
Dscoped_allocator_ops_test.cc181 OpOutputList output_list; in ExecOp()
303 OpOutputList output_list; in ExecOp()
Dragged_tensor_from_variant_op.cc267 OpOutputList splits_out; in ReturnRaggedTensor()
Dragged_gather_op.cc219 OpOutputList splits_out; in WriteSplits()
Ddecode_csv_op.cc76 OpOutputList output; in Compute()
Dbarrier_ops.cc613 OpOutputList values_output; in ComputeAsync()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h484 class OpOutputList {
486 typedef OpArgIterator<OpOutputList, const Tensor*> Iterator;
487 OpOutputList() : ctx_(nullptr), start_(0), stop_(0) {} in OpOutputList() function
488 OpOutputList(OpKernelContext* ctx, int start, int stop) in OpOutputList() function
490 OpOutputList& operator=(const OpOutputList& other) = default;
930 Status output_list(StringPiece name, OpOutputList* list);
1692 inline Tensor* OpOutputList::operator[](int i) {
1698 inline bool OpOutputList::required(int i) const { in required()
1704 inline DataType OpOutputList::expected_output_dtype(int i) const { in expected_output_dtype()
1710 inline Status OpOutputList::allocate(int i, const TensorShape& shape, in allocate()
[all …]
Dop_kernel_test.cc730 TEST_F(OpKernelBuilderTest, OpOutputList) { in TEST_F() argument
747 OpOutputList out_list; in TEST_F()
Dop_kernel.cc641 Status OpKernelContext::output_list(StringPiece name, OpOutputList* list) { in output_list()
644 *list = OpOutputList(this, start, stop); in output_list()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc168 OpOutputList summaries_output_list; in Compute()
235 OpOutputList summaries_output_list; in Compute()
467 OpOutputList bucket_boundaries_list; in Compute()
526 OpOutputList buckets_list; in Compute()
Dstats_ops.cc85 OpOutputList output_node_ids_list; in Compute()
88 OpOutputList output_gains_list; in Compute()
91 OpOutputList output_thresholds_list; in Compute()
94 OpOutputList output_left_node_contribs_list; in Compute()
98 OpOutputList output_right_node_contribs_list; in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_defun_op.cc55 OpOutputList output TF_GUARDED_BY(mu);