Home
last modified time | relevance | path

Searched defs:new_layout (Results 1 – 25 of 25) sorted by relevance

/external/rust/crates/bytes/tests/
Dtest_bytes_odd_alloc.rs21 let new_layout = match Layout::from_size_align(size, 1) { in alloc() localVariable
39 let new_layout = match Layout::from_size_align(size, 1) { in dealloc() localVariable
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dreplicated_spmd_expander.cc51 Layout new_layout = in ReplicatedRelayoutOperandsAndOutputs() local
70 Layout new_layout = in ReplicatedRelayoutOperandsAndOutputs() local
Dqr_spmd_expander.cc50 Layout new_layout = in ExpandOp() local
/external/tensorflow/tensorflow/dtensor/python/
Ddtensor_device.py182 def copy_to_mesh(self, tensor, new_layout, source_layout=None) -> ops.Tensor: argument
/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc482 Layout new_layout = LayoutUtil::MoveDimToMajor(layout, 0); in TEST_F() local
Dshape_util.cc1179 Layout* new_layout = new_shape.mutable_layout(); in PermuteDimensions() local
Dliteral.cc849 Literal LiteralBase::Relayout(const Layout& new_layout, in Relayout()
/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/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device_util.h267 virtual void UpdateLayout(const Layout& new_layout, TF_Status* status) { in UpdateLayout()
Ddtensor_device_util.cc408 void ResourceHandleWithLayout::UpdateLayout(const Layout& new_layout, in UpdateLayout()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization_test.cc595 Layout new_layout = layout; in ChooseCompactLayoutForShape() local
Dlayout_assignment.cc2282 Layout new_layout = in RunOnComputation() local
Dhlo_instruction.cc4542 void HloInstruction::RelayoutConstant(const Layout& new_layout, in RelayoutConstant()
Dhlo_instructions.cc1472 void HloConstantInstruction::RelayoutConstant(const Layout& new_layout, in RelayoutConstant()
/external/mesa3d/src/gallium/drivers/zink/
Dzink_context.c861 VkImageAspectFlags aspect, VkImageLayout new_layout) in zink_resource_barrier()
/external/tensorflow/tensorflow/dtensor/mlir/
Dlayout_propagation_v2.cc505 const auto new_layout = Layout::GetLayout(merged_spec, layouts[0].mesh()); in GetMostShardedLayout() local
/external/vulkan-validation-layers/tests/
Dvklayertests_command.cpp3825 …&image_barrier](const VkCommandBufferObj &cb, VkImageLayout old_layout, VkImageLayout new_layout) { in TEST_F()
Dvkpositivelayertests.cpp3680 VkImageLayout new_layout; in TEST_F() member
/external/rust/crates/ash/src/vk/
Ddefinitions.rs3141 pub new_layout: ImageLayout, field
3201 pub fn new_layout(mut self, new_layout: ImageLayout) -> Self { in new_layout() method
45779 pub new_layout: ImageLayout, field
45849 pub fn new_layout(mut self, new_layout: ImageLayout) -> Self { in new_layout() method
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_510.h106994 __u32 new_layout; member
107032 __le32 new_layout; member
107179 int new_layout; member
Dvmlinux.h106994 __u32 new_layout; member
107032 __le32 new_layout; member
107179 int new_layout; member
/external/bcc/libbpf-tools/arm64/
Dvmlinux.h91145 __u32 new_layout; member
91183 __le32 new_layout; member
91330 int new_layout; member
Dvmlinux_510.h91145 __u32 new_layout; member
91183 __le32 new_layout; member
91330 int new_layout; member
/external/bcc/libbpf-tools/x86/
Dvmlinux_505.h96571 __u32 new_layout; member
96609 __le32 new_layout; member
96723 int new_layout; member
Dvmlinux.h96571 __u32 new_layout; member
96609 __le32 new_layout; member
96723 int new_layout; member