| /external/tensorflow/tensorflow/compiler/xla/ |
| index_util.cc | 59 // linear_index =
64 // linear_index =
79 int64 linear_index = 0; local
84 linear_index = multi_index[dimension];
88 linear_index += scale * multi_index[dimension];
92 return linear_index;
96 const Shape& shape, int64 linear_index) {
99 DCHECK_GE(linear_index, 0);
100 DCHECK_LT(linear_index, ShapeUtil::ElementsIn(shape));
105 // I{L(0)} = linear_index % D{L(0)
[all...] |
| index_util.h | 44 const Shape& shape, int64 linear_index);
|
| index_util_test.cc | 133 for (auto linear_index : linear_indexes) {
135 IndexUtil::LinearIndexToMultidimensionalIndex(shape, linear_index);
136 EXPECT_EQ(linear_index, IndexUtil::MultidimensionalIndexToLinearIndex(
|
| literal_util.cc | 197 auto linear_index = [](const Shape& shape, local
207 StridedCopy(data<NativeT>(), linear_index(shape(), dest_base), 0,
209 linear_index(src_literal.shape(), src_base), 0, 1);
234 int64 src_index = linear_index(src_literal.shape(), src_indexes);
235 int64 dest_index = linear_index(shape(), dest_indexes);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| parallel_loop_emitter.cc | 56 // linear_index = blockIdx.x * blockDim.x + threadIdx.x;
57 // if (linear_index < num_elements) {
58 // array_index = LinearIndexToMultidimensionalIndex(shape_, linear_index);
84 llvm::Value* linear_index = ir_builder_->CreateAdd( local
89 thread_id, "linear_index", /*HasNUW=*/true, /*HasNSW=*/true);
91 // Add an @llvm.assume(linear_index < threads_per_block * num_blocks).
94 // currently determine the range of linear_index precisely. InstCombine uses
102 linear_index,
110 linear_index, ir_builder_->getInt64(ShapeUtil::ElementsIn(shape_))),
119 return llvm_ir::IrArray::Index(linear_index, shape_, ir_builder_)
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| literal_test_util.cc | 489 const Shape& shape, const int64 linear_index) {
496 max_rel_linear_index_ = linear_index;
500 max_abs_linear_index_ = linear_index;
507 linear_index))
516 IndexUtil::LinearIndexToMultidimensionalIndex(shape, linear_index);
527 first_linear_index_ = linear_index;
530 last_linear_index_ = linear_index;
531 miscompares_.data<bool>()[linear_index] = true;
672 const int64 linear_index) {
674 static_cast<float>(actual), shape, linear_index); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | 80 bool SetValueInLiteral(int64 value, int64 linear_index, Literal* literal);
81 bool SetValueInLiteral(double value, int64 linear_index, Literal* literal);
82 bool SetValueInLiteral(bool value, int64 linear_index, Literal* literal);
84 bool SetValueInLiteralHelper(ParsedElemT value, int64 linear_index,
1444 int64 linear_index = 0; local
[all...] |
