Searched defs:SetSameShapePolicy (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ | ||
| H A D | dtensor_device.cc | 157 void SetSameShapePolicy(bool enabled) { in SetSameShapePolicy() function in tensorflow::dtensor::DTensorDevice |