Home
last modified time | relevance | path

Searched refs:execution_functions (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc346 const ExecutionFunctions** execution_functions,
1259 const Graph& graph, ExecutionFunctions* execution_functions) { in AddExecutionFunctionDefsToFunctionDefLibrary() argument
1266 for (TranslatedFunction& function : execution_functions->function_list) { in AddExecutionFunctionDefsToFunctionDefLibrary()
1316 const int num_outputs, const ExecutionFunctions** execution_functions, in LowerToSPMDFunction() argument
1358 *execution_functions = cache_key_and_func.second; in LowerToSPMDFunction()
1359 if (*execution_functions != nullptr) { in LowerToSPMDFunction()
1388 *execution_functions = function_manager_.AddCachedFunction( in LowerToSPMDFunction()
1451 *execution_functions = function_manager_.AddCachedFunction( in LowerToSPMDFunction()
1495 const ExecutionFunctions* execution_functions = nullptr; in ExecuteRegularOperation() local
1498 &execution_functions, status); in ExecuteRegularOperation()
[all …]
H A Ddtensor_device_util.cc849 ExecutionFunctions execution_functions; in IdentifyAllFunctionsToExecute() local
850 execution_functions.function_list = std::vector<TranslatedFunction>(); in IdentifyAllFunctionsToExecute()
923 execution_functions.function_list.emplace_back(std::move(function)); in IdentifyAllFunctionsToExecute()
926 if (execution_functions.function_list.empty()) { in IdentifyAllFunctionsToExecute()
932 return execution_functions; in IdentifyAllFunctionsToExecute()