Home
last modified time | relevance | path

Searched refs:GetMergedOperandLayout (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Delementwise_spmd_expander.cc126 GetMergedOperandLayout(input_layouts, op)); in ComputeLayoutForward()
H A Dconcat_spmd_expander.cc112 GetMergedOperandLayout(concat_operands_layouts, op)); in ComputeLayoutForward()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expander_common.h79 StatusOr<absl::optional<Layout>> GetMergedOperandLayout(
H A Dspmd_expander_common.cc277 StatusOr<absl::optional<Layout>> GetMergedOperandLayout( in GetMergedOperandLayout() function