Searched defs:inputDim (Results 1 – 5 of 5) sorted by relevance
137 auto inputDim = b->create<arith::AddIOp>( in elementalLower() local168 int inputDim = std::distance(broadcastDimensions.begin(), it); in elementalLowerImplForBroadcastInDimOps() local
78 int inputDim = 3) in NCCLTest()247 int inputDim) in SparseAllreduceNCCLTest()429 const int inputDim = 3; in testSparseAllreduce() local480 const int inputDim = 2500; in testSparseAllreduceLarge() local
601 int64_t inputDim = inputDims[permutationRef[i]]; in verify() local
4724 static int64_t inferSliceDim(int64_t inputDim, int64_t start, int64_t end, in inferSliceDim()
6796 static int64_t inferSliceDim(int64_t inputDim, int64_t start, int64_t end, in inferSliceDim()