Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc426 std::vector<Tensor> input_element; in CallFunction() local
451 [this, ctx, result](std::vector<Tensor> input_element) { in CallFunction()
Dcaptured_function.cc448 const std::vector<Tensor>& input_element, int64 thread_index, in MakeIteratorFromInputElement()
458 const std::vector<Tensor>& input_element, int64 thread_index, in MakeIteratorFromInputElement()
Dshuffle_dataset_op.cc180 std::vector<Tensor> input_element; in GetNextInternal() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dparse_example_dataset_op.cc583 std::vector<Tensor> input_element; in CallFunction() local
599 [this, ctx, result](std::vector<Tensor> input_element) { in CallFunction()
Dmap_and_batch_dataset_op.cc401 std::vector<Tensor> input_element; in CallFunction() local