Home
last modified time | relevance | path

Searched defs:parallel_inputs (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc1458 const std::vector<parallel_device::ParallelTensor*>& parallel_inputs, in ExecuteFunctionAndWait()
1601 StatusOr<std::vector<parallel_device::ParallelTensor*>> parallel_inputs = in ExecuteRegularOperation() local
1661 std::vector<parallel_device::ParallelTensor*> parallel_inputs; in ExecuteRegularOperation() local
H A Ddtensor_device_util.cc1010 std::vector<parallel_device::ParallelTensor*> parallel_inputs; in PrepareEmbeddingInputs() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device.cc143 std::vector<ParallelTensor*> parallel_inputs; in ExecuteWithSpecialOps() local