Home
last modified time | relevance | path

Searched refs:ExtractRequiredLayoutFromOperands (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dtensorlist_getitem_spmd_expander.cc28 ExtractRequiredLayoutFromOperands(op)); in ExpandOp()
H A Dtensorlist_setitem_spmd_expander.cc29 ExtractRequiredLayoutFromOperands(op)); in ExpandOp()
H A Dqr_spmd_expander.cc46 ExtractRequiredLayoutFromOperands(op)); in ExpandOp()
H A Dreplicated_spmd_expander.cc101 ExtractRequiredLayoutFromOperands(op)); in ExpandOp()
H A Ddataparallel_spmd_expander.cc284 ExtractRequiredLayoutFromOperands(op)); in ExpandOp()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dlayout_parsing.h66 StatusOr<std::vector<Layout>> ExtractRequiredLayoutFromOperands(
H A Dlayout_parsing.cc217 StatusOr<std::vector<Layout>> ExtractRequiredLayoutFromOperands( in ExtractRequiredLayoutFromOperands() function