Searched refs:ExecutionFunctions (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device_util.h | 98 struct ExecutionFunctions { struct
544 const ExecutionFunctions* AddCachedFunction(const DTensorOperation& op,
546 ExecutionFunctions function);
552 std::pair<tensorflow::Fprint128, const ExecutionFunctions*> GetCachedFunction(
579 absl::flat_hash_map<tensorflow::Fprint128, ExecutionFunctions,
609 StatusOr<ExecutionFunctions> IdentifyAllFunctionsToExecute(
|
| H A D | dtensor_device_util.cc | 555 std::pair<tensorflow::Fprint128, const ExecutionFunctions*>
566 return std::pair<Fprint128, ExecutionFunctions*>(cache_key, &iter->second); in GetCachedFunction()
614 const ExecutionFunctions* FunctionManager::AddCachedFunction( in AddCachedFunction()
616 ExecutionFunctions function) { in AddCachedFunction()
846 StatusOr<ExecutionFunctions> IdentifyAllFunctionsToExecute( in IdentifyAllFunctionsToExecute()
849 ExecutionFunctions execution_functions; in IdentifyAllFunctionsToExecute()
|
| H A D | dtensor_device.cc | 82 StatusOr<ExecutionFunctions> ABSL_ATTRIBUTE_WEAK PipeliningPartitionerRun( in PipeliningPartitionerRun()
346 const ExecutionFunctions** execution_functions,
1259 const Graph& graph, ExecutionFunctions* execution_functions) { in AddExecutionFunctionDefsToFunctionDefLibrary()
1316 const int num_outputs, const ExecutionFunctions** execution_functions, in LowerToSPMDFunction()
1355 std::pair<tensorflow::Fprint128, const ExecutionFunctions*> in LowerToSPMDFunction()
1382 ExecutionFunctions functions, in LowerToSPMDFunction()
1425 ExecutionFunctions functions, in LowerToSPMDFunction()
1495 const ExecutionFunctions* execution_functions = nullptr; in ExecuteRegularOperation()
|