Searched defs:output_dynamic_dims (Results 1 – 1 of 1) sorted by relevance
318 absl::Span<HloInstruction*> output_dynamic_dims, HloInstruction* one, in GenerateBinaryMask()471 absl::Span<HloInstruction*> output_dynamic_dims, in RewriteDynamicReshapeSplitInput()773 absl::Span<HloInstruction*> output_dynamic_dims, in RewriteDynamicReshapeSingleGroup()1657 std::vector<HloInstruction*> output_dynamic_dims; in RewriteDynamicReshape() local