Searched refs:RelayoutOneHotInput (Results 1 – 1 of 1) sorted by relevance
960 Status RelayoutOneHotInput(const absl::optional<Layout>& input_layout, in RelayoutOneHotInput() function1005 RelayoutOneHotInput(input_layout, output_layout, axis, one_hot_op)); in ExpandOp()