Searched defs:dimno (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | mirror_pad_op.cc | 43 for (int64 dimno = original_shape.rank() - 1; dimno >= 0; --dimno) { in DoMirrorPad() local
129 for (int64 dimno = original_shape.rank() - 1; dimno >= 0; --dimno) { in DoMirrorPadGrad() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | index_util.cc | 118 for (int64 dimno = indices.size() - 1; dimno >= 0; --dimno) { in BumpIndices() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 980 int64 limit_index, int64 stride, int64 dimno) { in SliceInDim()
1093 XlaOp XlaBuilder::PadInDim(XlaOp operand, XlaOp padding_value, int64 dimno, in PadInDim()
4118 XlaOp PadInDim(XlaOp operand, XlaOp padding_value, int64 dimno, int64 pad_lo, in PadInDim()
4162 int64 stride, int64 dimno) { in SliceInDim()
|