HomeSort by relevance Sort by last modified time
    Searched refs:input_element (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/data/
dataset_utils.cc 23 IteratorContext* ctx, const std::vector<Tensor>& input_element,
29 captured_func->RunWithBorrowedArgs(ctx, input_element, &return_values));
dataset_utils.h 27 IteratorContext* ctx, const std::vector<Tensor>& input_element,
map_and_batch_dataset_op.cc 272 std::vector<Tensor> input_element; local
274 input_impl_->GetNext(ctx, &input_element, &result->end_of_input);
280 // Call `captured_func_(input_element)`, store the result in
285 IteratorContext* ctx, std::vector<Tensor> input_element) {
287 ctx, std::move(input_element), &result->return_values,
331 new IteratorContext(*ctx), std::move(input_element)));
parallel_map_dataset_op.cc 313 std::vector<Tensor> input_element; local
316 input_impl_->GetNext(ctx, &input_element, &end_of_input);
325 // Call `func_(input_element)`, store the result in
330 ctx, std::move(input_element), &result->return_values,
shuffle_dataset_op.cc 97 std::vector<Tensor> input_element; variable
100 TF_RETURN_IF_ERROR(input_impl_->GetNext(ctx, &input_element,
121 std::move(input_element);

Completed in 71 milliseconds