Searched refs:dimno (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | mirror_pad_op.cc | 41 for (int64 dimno = original_shape.rank() - 1; dimno >= 0; --dimno) { in DoMirrorPad() local 42 auto t_rev = xla::Rev(accum, {dimno}); in DoMirrorPad() 43 int64 lhs_padding = pad_literal.Get<int64>({dimno, 0}); in DoMirrorPad() 44 int64 rhs_padding = pad_literal.Get<int64>({dimno, 1}); in DoMirrorPad() 45 int64 dim_size = original_shape.dimensions(dimno); in DoMirrorPad() 56 dim_size - excluded_edges, 1, dimno); in DoMirrorPad() 58 excluded_edges + rhs_padding, 1, dimno); in DoMirrorPad() 59 accum = xla::ConcatInDim(b, {lhs_pad, accum, rhs_pad}, dimno); in DoMirrorPad()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | index_util.cc | 118 for (int64 dimno = indices.size() - 1; dimno >= 0; --dimno) { in BumpIndices() local 119 int64 limit = shape.dimensions(dimno); in BumpIndices() 120 if (indices[dimno] + 1 < limit) { in BumpIndices() 121 indices[dimno]++; in BumpIndices() 122 std::fill(indices.begin() + dimno + 1, indices.end(), 0); in BumpIndices()
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | local_computation_builder.cc | 524 int64 stride, int64 dimno) { in SliceInDim() argument 525 return xla::SliceInDim(operand.op(), start_index, limit_index, stride, dimno); in SliceInDim()
|
D | local_computation_builder.h | 267 int64 limit_index, int64 stride, int64 dimno);
|
D | xla_client.py | 1337 def SliceInDim(self, operand, start_index, limit_index, stride, dimno): argument 1351 dimno)
|
D | xla_client_test.py | 1060 dimno=1) 1067 dimno=0)
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 744 int64 limit_index, int64 stride, int64 dimno) { in SliceInDim() argument 751 starts[dimno] = start_index; in SliceInDim() 752 limits[dimno] = limit_index; in SliceInDim() 753 strides[dimno] = stride; in SliceInDim() 2876 int64 stride, int64 dimno) { in SliceInDim() argument 2878 stride, dimno); in SliceInDim()
|
D | xla_builder.h | 351 int64 stride, int64 dimno); 739 int64 limit_index, int64 stride, int64 dimno); 1202 int64 stride, int64 dimno);
|