Searched defs:target_layout (Results 1 – 9 of 9) sorted by relevance
132 const Layout& mask_layout, const Layout& target_layout) { in MergeLayouts()294 const Layout target_layout = in ExpandOp() local
209 Layout target_layout = dtensor_send.target_layout(); in LowerDTensorSendFromCPUToTFOp() local
203 const Layout& target_layout = layout_or_status.ValueOrDie(); in LowerToSendRecv() local
278 const Layout target_layout = Layout::ReplicatedOnMesh(mesh, 0); in CloneEmptyIfWithPredicate() local
692 const Layout target_layout = *target_layout_or; in CopyToMesh() local815 StatusOr<Layout> target_layout = Layout::FromString(string_layout); in Pack() local919 StatusOr<Layout> target_layout = Layout::FromString(string_layout); in SparsePack() local
243 WORD32 target_layout; member
263 WORD32 target_layout; member
5261 VkImageLayout target_layout = subpass->attachments[i].layout; in cmd_buffer_begin_subpass() local5947 VkImageLayout target_layout = in cmd_buffer_end_subpass() local
674 WORD32 target_layout; member