Searched defs:dimno (Results 1 – 4 of 4) sorted by relevance
43 for (int64_t dimno = original_shape.rank() - 1; dimno >= 0; --dimno) { in DoMirrorPad() local130 for (int64_t dimno = original_shape.rank() - 1; dimno >= 0; --dimno) { in DoMirrorPadGrad() local
118 for (int64_t dimno = indices.size() - 1; dimno >= 0; --dimno) { in BumpIndices() local
149 for (int dimno = indices.size() - 1; dimno >= 0; --dimno) { in BumpIndices() local
991 int64_t dimno) { in SliceInDim()1104 XlaOp XlaBuilder::PadInDim(XlaOp operand, XlaOp padding_value, int64_t dimno, in PadInDim()3930 XlaOp PadInDim(XlaOp operand, XlaOp padding_value, int64_t dimno, in PadInDim()3974 int64_t stride, int64_t dimno) { in SliceInDim()