| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| loops.h | 64 StatusOr<std::vector<XlaOp>> ForEachIndex(
|
| loops.cc | 82 StatusOr<std::vector<XlaOp>> ForEachIndex(
|
| qr.cc | 226 TF_ASSIGN_OR_RETURN(auto values, ForEachIndex(std::min(m, n), S32, qr_body_fn,
302 ForEachIndex(n - 1, S32, body_fn, {w, y, vs, taus}, "wy", builder));
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| cholesky_expander.cc | 132 ForEachIndex(n, S32, body_fn, {a, l}, "unblocked", builder));
|
| hlo_evaluator_typed_visitor.h | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| shape_util.h | 654 static void ForEachIndex(const Shape& shape, absl::Span<const int64> base,
679 static void ForEachIndex(const Shape& shape, const FnType& visitor_function) {
685 // A parallel version of ForEachIndex(WithStatus). This can only be used if
[all...] |
| literal_test.cc | [all...] |
| shape_util_test.cc | 541 TEST(ShapeUtilTest, ForEachIndex) {
562 ShapeUtil::ForEachIndex(shape, zero_base, data.dimensions, step,
[all...] |
| literal.h | 712 // a ShapeUtil::ForEachIndex() scan across two literals.
[all...] |
| literal.cc | 261 ShapeUtil::ForEachIndex(src_literal.shape(), stride_config.base,
623 // every iteration of ShapeUtil::ForEachIndex.
631 ShapeUtil::ForEachIndex(
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| random_ops.cc | 186 xla::ForEachIndex(n, xla::S32, swap_body_fn, {swaps, indices},
|
