Home
last modified time | relevance | path

Searched refs:linear_index (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dindex_util.cc79 int64 linear_index = 0; in MultidimensionalIndexToLinearIndex() local
84 linear_index = multi_index[dimension]; in MultidimensionalIndexToLinearIndex()
88 linear_index += scale * multi_index[dimension]; in MultidimensionalIndexToLinearIndex()
92 return linear_index; in MultidimensionalIndexToLinearIndex()
96 const Shape& shape, int64 linear_index) { in LinearIndexToMultidimensionalIndex() argument
99 DCHECK_GE(linear_index, 0); in LinearIndexToMultidimensionalIndex()
100 DCHECK_LT(linear_index, ShapeUtil::ElementsIn(shape)); in LinearIndexToMultidimensionalIndex()
115 (linear_index / divisor) % shape.dimensions(dimension); in LinearIndexToMultidimensionalIndex()
Dindex_util_test.cc133 for (auto linear_index : linear_indexes) { in TEST() local
135 IndexUtil::LinearIndexToMultidimensionalIndex(shape, linear_index); in TEST()
136 EXPECT_EQ(linear_index, IndexUtil::MultidimensionalIndexToLinearIndex( in TEST()
Dindex_util.h44 const Shape& shape, int64 linear_index);
Dliteral_util.cc197 auto linear_index = [](const Shape& shape, in CopySliceFromInternal() local
207 StridedCopy(data<NativeT>(), linear_index(shape(), dest_base), 0, in CopySliceFromInternal()
209 linear_index(src_literal.shape(), src_base), 0, 1); in CopySliceFromInternal()
234 int64 src_index = linear_index(src_literal.shape(), src_indexes); in CopySliceFromInternal()
235 int64 dest_index = linear_index(shape(), dest_indexes); in CopySliceFromInternal()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dparallel_loop_emitter.cc84 llvm::Value* linear_index = ir_builder_->CreateAdd( in EmitIndexAndSetExitBasicBlock() local
102 linear_index, in EmitIndexAndSetExitBasicBlock()
110 linear_index, ir_builder_->getInt64(ShapeUtil::ElementsIn(shape_))), in EmitIndexAndSetExitBasicBlock()
119 return llvm_ir::IrArray::Index(linear_index, shape_, ir_builder_); in EmitIndexAndSetExitBasicBlock()
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
Dhlo_parser.cc80 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,
1266 bool HloParser::SetValueInLiteral(int64 value, int64 linear_index, in SetValueInLiteral() argument
1271 return SetValueInLiteralHelper<int8>(value, linear_index, literal); in SetValueInLiteral()
1273 return SetValueInLiteralHelper<int16>(value, linear_index, literal); in SetValueInLiteral()
1275 return SetValueInLiteralHelper<int32>(value, linear_index, literal); in SetValueInLiteral()
1277 return SetValueInLiteralHelper<int64>(value, linear_index, literal); in SetValueInLiteral()
1279 return SetValueInLiteralHelper<uint8>(value, linear_index, literal); in SetValueInLiteral()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tests/
Dliteral_test_util.cc489 const Shape& shape, const int64 linear_index) { in UpdateAndLogMiscompares() argument
496 max_rel_linear_index_ = linear_index; in UpdateAndLogMiscompares()
500 max_abs_linear_index_ = linear_index; in UpdateAndLogMiscompares()
507 linear_index)) in UpdateAndLogMiscompares()
516 IndexUtil::LinearIndexToMultidimensionalIndex(shape, linear_index); in UpdateAndLogMiscompares()
527 first_linear_index_ = linear_index; in UpdateAndLogMiscompares()
530 last_linear_index_ = linear_index; in UpdateAndLogMiscompares()
531 miscompares_.data<bool>()[linear_index] = true; in UpdateAndLogMiscompares()
672 const int64 linear_index) { in UpdateAndLogMiscompares() argument
674 static_cast<float>(actual), shape, linear_index); in UpdateAndLogMiscompares()
[all …]