Searched refs:input_segment_end (Results 1 – 1 of 1) sorted by relevance
569 llvm::SmallVectorImpl<int64_t>& input_segment_end, in ComputeReshapeSegments() argument602 input_segment_end.emplace_back(input_offset); in ComputeReshapeSegments()680 llvm::SmallVector<int64_t, 4> input_segment_end; in ExpandOp() local690 input_segment_start, input_segment_end, in ExpandOp()717 const int prev_input_segment_end = (i == 0 ? 0 : input_segment_end[i - 1]); in ExpandOp()750 for (int j = input_start + 1; j < input_segment_end[i]; ++j) in ExpandOp()760 for (int j = input_segment_end.back(); j < tgt_input_layout.size(); ++j) in ExpandOp()816 llvm::SmallVector<int64_t, 4> input_segment_end; in ComputeLayoutForward() local824 input_segment_end, output_segment_start, in ComputeLayoutForward()849 llvm::SmallVector<int64_t, 4> input_segment_end; in ComputeLayoutBackward() local[all …]