Home
last modified time | relevance | path

Searched refs:dimno (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmirror_pad_op.cc41 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/
Dindex_util.cc118 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/
Dlocal_computation_builder.cc524 int64 stride, int64 dimno) { in SliceInDim() argument
525 return xla::SliceInDim(operand.op(), start_index, limit_index, stride, dimno); in SliceInDim()
Dlocal_computation_builder.h267 int64 limit_index, int64 stride, int64 dimno);
Dxla_client.py1337 def SliceInDim(self, operand, start_index, limit_index, stride, dimno): argument
1351 dimno)
Dxla_client_test.py1060 dimno=1)
1067 dimno=0)
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc744 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()
Dxla_builder.h351 int64 stride, int64 dimno);
739 int64 limit_index, int64 stride, int64 dimno);
1202 int64 stride, int64 dimno);