Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dsoftmax_spmd_expander.cc372 mlir::Operation* op, bool is_sparse, const Layout& features_layout, in MaybeRelayoutInputs()
633 absl::optional<Layout> features_layout; in ComputeLayoutForward() local