Home
last modified time | relevance | path

Searched refs:new_layout (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test.cc570 Layout new_layout = layout; in ChooseCompactLayoutForShape() local
571 new_layout.set_minor_to_major(0, second_minor_index); in ChooseCompactLayoutForShape()
572 new_layout.set_minor_to_major(1, most_minor_index); in ChooseCompactLayoutForShape()
573 *result.mutable_layout() = new_layout; in ChooseCompactLayoutForShape()
Dhlo_instructions.cc1143 void HloConstantInstruction::RelayoutConstant(const Layout& new_layout, in RelayoutConstant() argument
1153 !LayoutUtil::Equal(mutable_array_subshape->layout(), new_layout)) { in RelayoutConstant()
1154 *literal_ = literal_->Relayout(new_layout, shape_index); in RelayoutConstant()
1155 *mutable_array_subshape->mutable_layout() = new_layout; in RelayoutConstant()
Dlayout_assignment.cc1875 Layout new_layout = in RunOnComputation() local
1881 TF_RETURN_IF_ERROR(constraints.SetBufferLayout(new_layout, buffer, in RunOnComputation()
Dhlo_instruction.h1606 void RelayoutConstant(const Layout& new_layout,
Dhlo_instructions.h734 void RelayoutConstant(const Layout& new_layout,
Dhlo_instruction.cc3612 void HloInstruction::RelayoutConstant(const Layout& new_layout, in RelayoutConstant() argument
3614 Cast<HloConstantInstruction>(this)->RelayoutConstant(new_layout, shape_index); in RelayoutConstant()
/external/kernel-headers/original/uapi/linux/raid/
Dmd_p.h185 __u32 new_layout; /* 16 new layout */ member
263 __le32 new_layout; /* new layout */ member
/external/mesa3d/src/gallium/drivers/zink/
Dzink_context.h168 VkImageAspectFlags aspect, VkImageLayout new_layout);
Dzink_context.c861 VkImageAspectFlags aspect, VkImageLayout new_layout) in zink_resource_barrier() argument
873 access_dst_flags(new_layout), in zink_resource_barrier()
875 new_layout, in zink_resource_barrier()
884 pipeline_dst_stage(new_layout), in zink_resource_barrier()
891 res->layout = new_layout; in zink_resource_barrier()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc971 Layout* new_layout = new_shape.mutable_layout(); in PermuteDimensions() local
972 new_layout->set_format(DENSE); in PermuteDimensions()
973 new_layout->clear_minor_to_major(); in PermuteDimensions()
976 new_layout->add_minor_to_major(index); in PermuteDimensions()
Dliteral.cc572 Literal LiteralBase::Relayout(const Layout& new_layout, in Relayout() argument
577 TF_CHECK_OK(LayoutUtil::ValidateLayoutForShape(new_layout, *subshape)); in Relayout()
578 *subshape->mutable_layout() = new_layout; in Relayout()
Dliteral.h273 Literal Relayout(const Layout& new_layout,
/external/vulkan-validation-layers/tests/
Dvktestbinding.h506 … VkImageLayout new_layout, const VkImageSubresourceRange &range,
514 barrier.newLayout = new_layout;
Dvkpositivelayertests.cpp3680 VkImageLayout new_layout; in TEST_F() member
3703 img_barrier.newLayout = buffer_layouts[i].new_layout; in TEST_F()
3707 img_barrier.oldLayout = buffer_layouts[i].new_layout; in TEST_F()
Dvklayertests_command.cpp3825 …&image_barrier](const VkCommandBufferObj &cb, VkImageLayout old_layout, VkImageLayout new_layout) { in TEST_F() argument
3827 image_barrier.newLayout = new_layout; in TEST_F()
/external/vulkan-validation-layers/layers/
Dcore_validation.h1067 VkImageLayout new_layout) const;
Dcore_validation.cpp10256 VkImageLayout new_layout) const { in AddAttachmentUse()
10266 if (attachment_layouts[attachment] != new_layout) { in AddAttachmentUse()
10270 …ttachment, string_VkImageLayout(attachment_layouts[attachment]), string_VkImageLayout(new_layout)); in AddAttachmentUse()
10280 attachment_layouts[attachment] = new_layout; in AddAttachmentUse()