Home
last modified time | relevance | path

Searched defs:target_layout (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Ddtensor_op_spmd_expander.cc132 const Layout& mask_layout, const Layout& target_layout) { in MergeLayouts()
294 const Layout target_layout = in ExpandOp() local
/external/tensorflow/tensorflow/dtensor/mlir/
Ddtensor_send_recv.cc209 Layout target_layout = dtensor_send.target_layout(); in LowerDTensorSendFromCPUToTFOp() local
Dhandle_cross_cluster_dependencies.cc203 const Layout& target_layout = layout_or_status.ValueOrDie(); in LowerToSendRecv() local
Dmerge_clusters.cc278 const Layout target_layout = Layout::ReplicatedOnMesh(mesh, 0); in CloneEmptyIfWithPredicate() local
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device.cc692 const Layout target_layout = *target_layout_or; in CopyToMesh() local
815 StatusOr<Layout> target_layout = Layout::FromString(string_layout); in Pack() local
919 StatusOr<Layout> target_layout = Layout::FromString(string_layout); in SparsePack() local
/external/libxaac/decoder/drc_src/
Dimpd_drc_common.h243 WORD32 target_layout; member
Dimpd_drc_struct.h263 WORD32 target_layout; member
/external/mesa3d/src/intel/vulkan/
DgenX_cmd_buffer.c5261 VkImageLayout target_layout = subpass->attachments[i].layout; in cmd_buffer_begin_subpass() local
5947 VkImageLayout target_layout = in cmd_buffer_end_subpass() local
/external/libxaac/decoder/
Dixheaacd_config.h674 WORD32 target_layout; member