Searched defs:input_dynamic_dims (Results 1 – 1 of 1) sorted by relevance
567 int64_t output_dim, absl::Span<HloInstruction*> input_dynamic_dims, in RewriteDynamicReshapeCombineInput()717 absl::Span<HloInstruction*> input_dynamic_dims, in RewriteDynamicReshapeSingleGroup()1585 std::vector<HloInstruction*> input_dynamic_dims; in RewriteDynamicReshape() local