Searched refs:function_manager_ (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device.cc | 405 FunctionManager function_manager_; member in tensorflow::dtensor::DTensorDevice
1336 function_manager_, inputs, doperation, *flib_def, eager_attributes, in LowerToSPMDFunction()
1356 cache_key_and_func = function_manager_.GetCachedFunction( in LowerToSPMDFunction()
1388 *execution_functions = function_manager_.AddCachedFunction( in LowerToSPMDFunction()
1397 function_manager_, inputs, doperation, *flib_def, eager_attributes, in LowerToSPMDFunction()
1451 *execution_functions = function_manager_.AddCachedFunction( in LowerToSPMDFunction()
|