Searched refs:linear_index (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/ |
D | index_util.cc | 79 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()
|
D | index_util_test.cc | 133 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()
|
D | index_util.h | 44 const Shape& shape, int64 linear_index);
|
D | literal_util.cc | 197 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/ |
D | parallel_loop_emitter.cc | 84 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/ |
D | 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, 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/ |
D | literal_test_util.cc | 489 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 …]
|