Searched refs:input_element (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | dataset_utils.cc | 23 IteratorContext* ctx, const std::vector<Tensor>& input_element, in MakeIteratorFromInputElement() argument 29 captured_func->RunWithBorrowedArgs(ctx, input_element, &return_values)); in MakeIteratorFromInputElement()
|
D | map_and_batch_dataset_op.cc | 272 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()
|
D | dataset_utils.h | 27 IteratorContext* ctx, const std::vector<Tensor>& input_element,
|
D | parallel_map_dataset_op.cc | 313 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()
|
D | shuffle_dataset_op.cc | 97 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()
|