| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | collectives.cc | 52 const dtensor::Layout& src_layout, const dtensor::Layout& tgt_layout, in EmitAllGather() argument
54 if (src_layout.IsEquivalent(tgt_layout)) return input; in EmitAllGather()
56 if (src_layout.rank() != tgt_layout.rank()) { in EmitAllGather()
59 src_layout.rank(), " vs ", tgt_layout.rank()); in EmitAllGather()
63 for (int i = 0; i < src_layout.rank(); ++i) { in EmitAllGather()
64 if (src_layout.sharding_spec(i) != tgt_layout.sharding_spec(i) && in EmitAllGather()
66 return errors::InvalidArgument("source layout (", src_layout.ToString(), in EmitAllGather()
88 GlobalTypeFromLocalType(src_layout, input_type)); in EmitAllGather()
96 mlir::dtensor::LayoutAttr::get(builder.getContext(), src_layout), in EmitAllGather()
199 mlir::Value input, const dtensor::Layout& src_layout, in EmitRelayout() argument
[all …]
|
| D | collectives.h | 35 const dtensor::Layout& src_layout, const dtensor::Layout& tgt_layout,
52 mlir::Value input, const dtensor::Layout& src_layout,
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuPermuteKernel.cpp | 124 const DataLayout src_layout = src->info()->data_layout(); in run_permute() local
131 …if((src_layout == DataLayout::NCHW && perm == PermutationVector{ 2U, 0U, 1U }) || (src_layout == D… in run_permute()
160 switch(src_layout) in run_permute()
192 if(src_layout == DataLayout::NCHW && perm == PermutationVector{ 2U, 0U, 1U }) in run_permute()
209 else if(src_layout == DataLayout::NHWC && perm == PermutationVector{ 1U, 2U, 0U }) in run_permute()
|
| /external/tensorflow/tensorflow/dtensor/mlir/utils/ |
| D | collective_lowering.cc | 443 const Layout src_layout = all_gather.input_layout(); in LowerAllGatherOp() local
447 for (int64 i = 0; i < src_layout.rank(); ++i) in LowerAllGatherOp()
448 if (src_layout.num_shards_for_dim(src_layout.dim(i)) > 1 && in LowerAllGatherOp()
489 const int64 num_devices = src_layout.num_devices(); in LowerAllGatherOp()
504 src_layout.device_location(device_id); in LowerAllGatherOp()
509 const int32 mesh_idx = src_layout.mesh() in LowerAllGatherOp()
510 .idx_for_dim(src_layout.sharding_spec(i)) in LowerAllGatherOp()
513 const int64 step = output_shape[i] / src_layout.num_shards()[i]; in LowerAllGatherOp()
535 {src_layout.mesh().min_global_device_id()}, builder, loc); in LowerAllGatherOp()
550 if (src_layout.mesh().is_tpu_mesh()) { in LowerAllGatherOp()
[all …]
|
| /external/mesa3d/src/gallium/frontends/lavapipe/ |
| D | lvp_private.h | 756 VkImageLayout src_layout; member
765 VkImageLayout src_layout; member
783 VkImageLayout src_layout; member
828 VkImageLayout src_layout; member
|
| D | lvp_descriptor_set.c | 419 const struct lvp_descriptor_set_binding_layout *src_layout = in lvp_UpdateDescriptorSets() local
422 &src->descriptors[src_layout->descriptor_index]; in lvp_UpdateDescriptorSets()
|
| D | lvp_cmd_buffer.c | 971 cmd->u.img_to_buffer.src_layout = srcImageLayout; in lvp_CmdCopyImageToBuffer()
1006 cmd->u.copy_image.src_layout = srcImageLayout; in lvp_CmdCopyImage()
1076 cmd->u.blit_image.src_layout = srcImageLayout; in lvp_CmdBlitImage()
1243 cmd->u.resolve_image.src_layout = srcImageLayout; in lvp_CmdResolveImage()
|
| /external/mesa3d/src/intel/vulkan/ |
| D | anv_descriptor_set.c | 1640 const struct anv_descriptor_set_binding_layout *src_layout = in anv_UpdateDescriptorSets() local
1643 &src->descriptors[src_layout->descriptor_index]; in anv_UpdateDescriptorSets()
1652 if (src_layout->data & ANV_DESCRIPTOR_INLINE_UNIFORM) { in anv_UpdateDescriptorSets()
1653 assert(src_layout->data == ANV_DESCRIPTOR_INLINE_UNIFORM); in anv_UpdateDescriptorSets()
1656 src->desc_mem.map + src_layout->descriptor_offset + in anv_UpdateDescriptorSets()
1663 unsigned desc_size = anv_descriptor_size(src_layout); in anv_UpdateDescriptorSets()
1668 src->desc_mem.map + src_layout->descriptor_offset + in anv_UpdateDescriptorSets()
|
| /external/mesa3d/src/intel/blorp/ |
| D | blorp_priv.h | 254 enum isl_msaa_layout src_layout; member
|
| D | blorp_blit.c | 1197 assert((key->src_layout == ISL_MSAA_LAYOUT_NONE) == in brw_blorp_build_nir_shader()
1310 key->tex_layout != key->src_layout) { in brw_blorp_build_nir_shader()
1312 key->src_layout); in brw_blorp_build_nir_shader()
1344 assert(key->tex_layout == key->src_layout); in brw_blorp_build_nir_shader()
1360 assert(key->tex_layout == key->src_layout); in brw_blorp_build_nir_shader()
1852 wm_prog_key->src_layout = params->src.surf.msaa_layout; in try_blorp_blit()
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_cmd_buffer.c | 56 VkImageLayout src_layout,
5939 VkImageLayout src_layout, in radv_handle_depth_image_transition() argument
5951 if (src_layout == VK_IMAGE_LAYOUT_UNDEFINED) { in radv_handle_depth_image_transition()
5953 …} else if (!radv_layout_is_htile_compressed(cmd_buffer->device, image, src_layout, src_render_loop… in radv_handle_depth_image_transition()
5956 …} else if (radv_layout_is_htile_compressed(cmd_buffer->device, image, src_layout, src_render_loop,… in radv_handle_depth_image_transition()
6067 VkImageLayout src_layout, in radv_init_color_image_metadata() argument
6125 VkImageLayout src_layout, in radv_handle_color_image_transition() argument
6133 if (src_layout == VK_IMAGE_LAYOUT_UNDEFINED) { in radv_handle_color_image_transition()
6135 src_layout, src_render_loop, in radv_handle_color_image_transition()
6143 if (src_layout == VK_IMAGE_LAYOUT_PREINITIALIZED) { in radv_handle_color_image_transition()
[all …]
|
| /external/vulkan-validation-layers/layers/ |
| D | descriptor_sets.cpp | 946 auto src_layout = src_set->GetLayout(); in ValidateCopyUpdate() local
960 if (src_layout->IsDestroyed()) { in ValidateCopyUpdate()
968 report_data->FormatHandle(src_layout->GetDescriptorSetLayout()).c_str()); in ValidateCopyUpdate()
1040 …if ((!VerifyUpdateConsistency(DescriptorSetLayout::ConstBindingIterator(src_layout.get(), update->… in ValidateCopyUpdate()
1049 …if ((src_layout->GetCreateFlags() & VK_DESCRIPTOR_SET_LAYOUT_CREATE_UPDATE_AFTER_BIND_POOL_BIT_EXT… in ValidateCopyUpdate()
1064 …if (!(src_layout->GetCreateFlags() & VK_DESCRIPTOR_SET_LAYOUT_CREATE_UPDATE_AFTER_BIND_POOL_BIT_EX… in ValidateCopyUpdate()
|