Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Ddtensor_op_spmd_expander.cc132 const Layout& mask_layout, const Layout& target_layout) { in MergeLayouts()
294 const Layout target_layout = in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_send_recv.cc209 Layout target_layout = dtensor_send.target_layout(); in LowerDTensorSendFromCPUToTFOp() local
H A Dhandle_cross_cluster_dependencies.cc203 const Layout& target_layout = layout_or_status.ValueOrDie(); in LowerToSendRecv() local
H A Dmerge_clusters.cc278 const Layout target_layout = Layout::ReplicatedOnMesh(mesh, 0); in CloneEmptyIfWithPredicate() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/impl/
H A DPacking.cpp301 api::GPUMemoryLayout target_layout, in channel_image_repacking()
/aosp_15_r20/external/libxaac/decoder/drc_src/
H A Dimpd_drc_common.h240 WORD32 target_layout; member
H A Dimpd_drc_struct.h263 WORD32 target_layout; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorConversions.cpp78 const Layout& target_layout, in dense_to_sparse_compressed_prepare_check_mask_values_batched()
/aosp_15_r20/external/libxaac/encoder/drc_src/
H A Dimpd_drc_uni_drc.h218 WORD32 target_layout; member
/aosp_15_r20/external/libxaac/decoder/
H A Dixheaacd_config.h676 WORD32 target_layout; member