Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dmeta_spmd_expander.cc960 Status RelayoutOneHotInput(const absl::optional<Layout>& input_layout, in RelayoutOneHotInput() function
1005 RelayoutOneHotInput(input_layout, output_layout, axis, one_hot_op)); in ExpandOp()