Searched defs:doperation (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device_util.cc | 527 const DTensorOperation& doperation, const NameAttrList& attributes, in CacheKeyForGraph()
557 const DTensorOperation& doperation, const NameAttrList& attributes, in GetCachedFunction()
621 bool FunctionManager::IsConstantFoldable(const DTensorOperation& doperation, in IsConstantFoldable()
654 const DTensorOperation& doperation, in PrepareGraphForMlir()
|
| H A D | dtensor_device.cc | 1315 const DTensorOperation& doperation, const TFE_OpAttrs* attributes, in LowerToSPMDFunction()
1493 const DTensorOperation& doperation, const TFE_OpAttrs* attributes, in ExecuteRegularOperation()
|