Searched defs:UpdateLayout (Results 1 – 5 of 5) sorted by relevance
167 virtual void UpdateLayout(Shape* shape) {} in UpdateLayout() function
87 bool BufferLayoutConstraint::UpdateLayout(int64_t priority, in UpdateLayout() function in xla::BufferLayoutConstraint2082 auto UpdateLayout = [this](const HloInstruction* operand, in CalculateComputationLayout() local
267 virtual void UpdateLayout(const Layout& new_layout, TF_Status* status) { in UpdateLayout() function
408 void ResourceHandleWithLayout::UpdateLayout(const Layout& new_layout, in UpdateLayout() function in tensorflow::dtensor::ResourceHandleWithLayout
558 "UpdateLayout":{ object