Searched defs:LocalShapeFromGlobalShape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ | ||
| H A D | tensor_layout.cc | 985 std::vector<int64_t> Layout::LocalShapeFromGlobalShape( in LocalShapeFromGlobalShape() function in tensorflow::dtensor::Layout |