Home
last modified time | relevance | path

Searched refs:input_segment_end (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dmeta_spmd_expander.cc569 llvm::SmallVectorImpl<int64_t>& input_segment_end, in ComputeReshapeSegments() argument
602 input_segment_end.emplace_back(input_offset); in ComputeReshapeSegments()
680 llvm::SmallVector<int64_t, 4> input_segment_end; in ExpandOp() local
690 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() local
824 input_segment_end, output_segment_start, in ComputeLayoutForward()
849 llvm::SmallVector<int64_t, 4> input_segment_end; in ComputeLayoutBackward() local
[all …]