Home
last modified time | relevance | path

Searched defs:output_segment_start (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dmeta_spmd_expander.cc570 llvm::SmallVectorImpl<int64_t>& output_segment_start, in ComputeReshapeSegments()
638 llvm::SmallVectorImpl<int64_t>& output_segment_start) { in MakeLayoutForReshape()
681 llvm::SmallVector<int64_t, 4> output_segment_start; in ExpandOp() local
817 llvm::SmallVector<int64_t, 4> output_segment_start; in ComputeLayoutForward() local
850 llvm::SmallVector<int64_t, 4> output_segment_start; in ComputeLayoutBackward() local