Home
last modified time | relevance | path

Searched refs:AddCachedFunction (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.h544 const ExecutionFunctions* AddCachedFunction(const DTensorOperation& op,
H A Ddtensor_device.cc1388 *execution_functions = function_manager_.AddCachedFunction( in LowerToSPMDFunction()
1451 *execution_functions = function_manager_.AddCachedFunction( in LowerToSPMDFunction()
H A Ddtensor_device_util.cc614 const ExecutionFunctions* FunctionManager::AddCachedFunction( in AddCachedFunction() function in tensorflow::dtensor::FunctionManager