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