Searched defs:output_dynamic_dims (Results 1 – 1 of 1) sorted by relevance
353 absl::Span<HloInstruction*> output_dynamic_dims, in RewriteDynamicReshapeSplitInput()718 absl::Span<HloInstruction*> output_dynamic_dims, in RewriteDynamicReshapeSingleGroup()1591 std::vector<HloInstruction*> output_dynamic_dims; in RewriteDynamicReshape() local