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
148 for (int dimno = indices.size() - 1; dimno >= 0; --dimno) { in BumpIndices() local
1047 int64_t dimno) { in SliceInDim()1160 XlaOp XlaBuilder::PadInDim(XlaOp operand, XlaOp padding_value, int64_t dimno, in PadInDim()4074 XlaOp PadInDim(XlaOp operand, XlaOp padding_value, int64_t dimno, in PadInDim()4118 int64_t stride, int64_t dimno) { in SliceInDim()