Home
last modified time | relevance | path

Searched refs:input_element (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc23 IteratorContext* ctx, const std::vector<Tensor>& input_element, in MakeIteratorFromInputElement() argument
29 captured_func->RunWithBorrowedArgs(ctx, input_element, &return_values)); in MakeIteratorFromInputElement()
Dmap_and_batch_dataset_op.cc272 std::vector<Tensor> input_element; in InvokeFunctionLocked() local
274 input_impl_->GetNext(ctx, &input_element, &result->end_of_input); in InvokeFunctionLocked()
285 IteratorContext* ctx, std::vector<Tensor> input_element) { in InvokeFunctionLocked() argument
287 ctx, std::move(input_element), &result->return_values, in InvokeFunctionLocked()
331 new IteratorContext(*ctx), std::move(input_element))); in InvokeFunctionLocked()
Ddataset_utils.h27 IteratorContext* ctx, const std::vector<Tensor>& input_element,
Dparallel_map_dataset_op.cc313 std::vector<Tensor> input_element; in InvokeFunctionLocked() local
316 input_impl_->GetNext(ctx, &input_element, &end_of_input); in InvokeFunctionLocked()
330 ctx, std::move(input_element), &result->return_values, in InvokeFunctionLocked()
Dshuffle_dataset_op.cc97 std::vector<Tensor> input_element; in GetNextInternal() local
100 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &input_element, in GetNextInternal()
121 std::move(input_element); in GetNextInternal()