Searched refs:ForEachIndex (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | tridiagonal.cc | 177 ForEachIndex(num_eqs - 1, S32, preparation_body_fn, in ThomasSolver()
222 ForEachIndex( in ThomasSolver()
267 ForEachIndex(num_eqs - 1, S32, bwd_reduction_fn, in ThomasSolver()
|
| D | loops.h | 64 StatusOr<std::vector<XlaOp>> ForEachIndex(
|
| D | loops.cc | 82 StatusOr<std::vector<XlaOp>> ForEachIndex( in ForEachIndex() function
|
| D | sorting.cc | 168 ForEachIndex(num_partitions - 1, S32, topk_body_fn, in TopKWithPartitions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | qr_expander.cc | 269 TF_ASSIGN_OR_RETURN(auto values, ForEachIndex(std::min(m, n), S32, qr_body_fn, in QrBlock()
332 ForEachIndex(n, S32, body_fn, {t, vtv}, "wy", builder)); in CompactWYRepresentation()
|
| D | cholesky_expander.cc | 120 ForEachIndex( in CholeskyUnblocked()
|
| D | hlo_evaluator_typed_visitor.h | 1592 ShapeUtil::ForEachIndex(
2684 ShapeUtil::ForEachIndex(iota->shape(), [&](absl::Span<const int64> idx) {
2950 ShapeUtil::ForEachIndex(update_literal.shape(), base,
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.h | 738 static void ForEachIndex(const Shape& shape, absl::Span<const int64> base, in ForEachIndex() function
763 static void ForEachIndex(const Shape& shape, const FnType& visitor_function) { in ForEachIndex() function
|
| D | literal_test.cc | 1053 ShapeUtil::ForEachIndex(source.shape(), zero_base, dimensions, step, in TEST_F()
1077 ShapeUtil::ForEachIndex(source.shape(), zero_base, copy_size, step, in TEST_F()
1269 ShapeUtil::ForEachIndex(literal.shape(), zero_base, data.dimensions, step, in TEST_F()
1311 ShapeUtil::ForEachIndex(literal.shape(), zero_base, data.dimensions, step, in TEST_F()
|
| D | shape_util_test.cc | 565 TEST(ShapeUtilTest, ForEachIndex) { in TEST() argument
586 ShapeUtil::ForEachIndex(shape, zero_base, data.dimensions, step, in TEST()
|
| D | literal.h | 1087 ShapeUtil::ForEachIndex( in PopulateInternal()
|
| D | literal.cc | 321 ShapeUtil::ForEachIndex(src_literal.shape(), stride_config.base, in CopySliceFromInternal()
801 ShapeUtil::ForEachIndex( in Broadcast()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 197 xla::ForEachIndex(n, xla::S32, swap_body_fn, {swaps, indices}, in Compile()
|