HomeSort by relevance Sort by last modified time
    Searched refs:SliceInMinorDims (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/client/lib/
slicing.h 31 XlaOp SliceInMinorDims(XlaOp x, absl::Span<const int64> start,
qr.cc 292 auto v = SliceInMinorDims(vs, {0}, {1});
293 auto beta = SliceInMinorDims(taus, {0}, {1});
354 auto a_block = SliceInMinorDims(a, {i, i}, {m, i + k});
367 auto a_panel = SliceInMinorDims(a, {i, i + k}, {m, n});
374 auto q_panel = SliceInMinorDims(q, {0, i}, {m, m});
385 q = SliceInMinorDims(q, {0, 0}, {m, p});
386 a = SliceInMinorDims(a, {0, 0}, {p, n});
matrix.cc 80 return SliceInMinorDims(result, {std::min<int64>(k, n)},
85 return SliceInMinorDims(result, {std::min<int64>(-k, m)},
slicing.cc 21 XlaOp SliceInMinorDims(XlaOp x, absl::Span<const int64> start,
svd.cc 757 Sort({-d, SliceInMinorDims(result.u, {0, 0}, {m, n})},
764 Sort({SliceInMinorDims(-d, {0, 0}, {n, n}), result.v},
783 SliceInMinorDims(result.u, {0, n}, {m, m})},
  /external/tensorflow/tensorflow/compiler/xla/service/
cholesky_expander.cc 179 auto lhs = SliceInMinorDims(l, {i, 0}, {n, i});
180 auto rhs = SliceInMinorDims(l, {i, 0}, {i + k, i});
182 auto before = SliceInMinorDims(a, {i, i}, {n, i + k});
187 auto x = SliceInMinorDims(a, {i, i}, {i + k, i + k});
194 auto panel = SliceInMinorDims(a, {i + k, i}, {n, i + k});
triangular_solve_expander.cc 94 SliceInMinorDims(a, {n - n % block_size, n - n % block_size}, {n, n});
288 MaybeConjugate(Collapse(SliceInMinorDims(inv_diag_blocks, {j, 0, 0},
301 auto b_row = SliceInMinorDims(b, start, end);
318 MaybeConjugate(SliceInMinorDims(a, start, end), conjugate_a);
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
xla_svd_op.cc 75 result.u = xla::SliceInMinorDims(result.u, {0, 0}, {m, p});
78 result.v = xla::SliceInMinorDims(result.v, {0, 0}, {n, p});

Completed in 1526 milliseconds