Home
last modified time | relevance | path

Searched defs:arg_layout (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/dtensor/mlir/
Dspmd_expansion.cc199 auto arg_layout = Layout::FromString(arg_layout_attr.getValue().str()); in UpdateFunctionArgsUsingLayout() local
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device_util.cc209 const auto& arg_layout = layouts[i]; in ParseAttrMap() local