Searched refs:AddCachedFunction (Results 1 – 3 of 3) sorted by relevance
544 const ExecutionFunctions* AddCachedFunction(const DTensorOperation& op,
1388 *execution_functions = function_manager_.AddCachedFunction( in LowerToSPMDFunction()1451 *execution_functions = function_manager_.AddCachedFunction( in LowerToSPMDFunction()
614 const ExecutionFunctions* FunctionManager::AddCachedFunction( in AddCachedFunction() function in tensorflow::dtensor::FunctionManager