Home
last modified time | relevance | path

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

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