Home
last modified time | relevance | path

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

12

/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/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/pigweed/pw_allocator/
Dlibc_allocator.cc34 void* LibCAllocator::DoReallocate(void* ptr, Layout new_layout) { in DoReallocate()
Dallocator.cc24 void* Allocator::DoReallocate(void* ptr, Layout new_layout) { in DoReallocate()
Dfallback_allocator_test.cc177 Layout new_layout = Layout::Of<uint32_t[3]>(); in TEST_F() local
190 Layout new_layout = Layout::Of<uint32_t[3]>(); in TEST_F() local
Dtest_harness.cc136 Layout new_layout = Layout(r.new_size, old.layout.alignment()); in HandleRequest() local
Dsynchronized_allocator_test.cc268 Layout new_layout = allocation.layout.Extend(1); in TestReallocate() local
/external/mesa3d/src/gallium/drivers/zink/
Dzink_synchronization.cpp163 zink_resource_image_needs_barrier(struct zink_resource *res, VkImageLayout new_layout, VkAccessFlag… in zink_resource_image_needs_barrier()
172 …rier_init(VkImageMemoryBarrier *imb, struct zink_resource *res, VkImageLayout new_layout, VkAccess… in zink_resource_image_barrier_init()
199 …er2_init(VkImageMemoryBarrier2 *imb, struct zink_resource *res, VkImageLayout new_layout, VkAccess… in zink_resource_image_barrier2_init()
511 …ge_barrier(struct zink_context *ctx, struct zink_resource *res, VkImageLayout new_layout, VkAccess… in zink_resource_image_barrier()
/external/crosvm/gpu_display/src/vulkan/
Dexternal_image.rs49 pub new_layout: ImageLayout, field
183 let new_layout = image_memory_barrier.new_layout; in release() localVariable
/external/pigweed/pw_allocator/public/pw_allocator/
Dsynchronized_allocator.h62 void* DoReallocate(void* ptr, Layout new_layout) override { in DoReallocate()
Dallocator.h115 void* Reallocate(void* ptr, Layout new_layout) { in Reallocate()
Dtracking_allocator.h137 Layout new_layout) { in DoReallocate()
/external/tensorflow/tensorflow/dtensor/python/
Ddtensor_device.py182 def copy_to_mesh(self, tensor, new_layout, source_layout=None) -> ops.Tensor: argument
/external/rust/crates/unsafe-libyaml/src/
Dlib.rs99 let new_layout = Layout::from_size_align(new_size, MALLOC_ALIGN) in realloc() localVariable
/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
/external/kernel-headers/original/uapi/linux/raid/
Dmd_p.h181 __u32 new_layout; /* 16 new layout */ member
259 __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
/external/rust/crates/vulkano/src/sync/
Dpipeline.rs2432 pub new_layout: ImageLayout, field
/external/mesa3d/src/vulkan/runtime/
Dvk_render_pass.c1248 VkImageLayout new_layout, in transition_image_range()
/external/tensorflow/tensorflow/dtensor/mlir/
Dlayout_propagation_v2.cc505 const auto new_layout = Layout::GetLayout(merged_spec, layouts[0].mesh()); in GetMostShardedLayout() local

12