Home
last modified time | relevance | path

Searched refs:getInt64 (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_support_library.h59 For(name, /*start=*/ir_builder_->getInt64(start), in For()
60 /*end=*/ir_builder_->getInt64(end), in For()
61 /*step=*/ir_builder_->getInt64(step), for_body_generator); in For()
90 /*step=*/ir_builder_->getInt64(step), peel_first_iteration, in For()
106 For(name, /*start=*/ir_builder_->getInt64(start), in For()
107 /*end=*/ir_builder_->getInt64(end), in For()
108 /*step=*/ir_builder_->getInt64(step), for_body_generator); in For()
Dtuple_ops.cc50 std::vector<llvm::Value*> element_index = {ir_builder->getInt64(0), in EmitTupleSelect()
51 ir_builder->getInt64(i)}; in EmitTupleSelect()
82 {ir_builder->getInt64(0), ir_builder->getInt64(i)})); in EmitTuple()
92 operand, {ir_builder->getInt64(0), ir_builder->getInt64(index)}); in EmitGetTupleElement()
Dir_array.cc56 auto* quot = ir_builder->CreateUDiv(linear, ir_builder->getInt64(divisor)); in Index()
59 quot, ir_builder->getInt64(size_of_current_dimension)); in Index()
150 llvm::Value* divisor = builder->getInt64(input_shape.dimensions(i)); in SourceIndexOfReshape()
152 source_multidim_index[i] = builder->getInt64(0); in SourceIndexOfReshape()
210 llvm::Value* logical_linear_index = builder->getInt64(0); in Linearize()
214 builder->CreateMul((*this)[i], builder->getInt64(multiplier), "", in Linearize()
241 actual_index.push_back(dim == 1 ? ir_builder->getInt64(0) : index[i]); in EmitArrayElementAddress()
260 std::vector<llvm::Value*> gep_indices(1, ir_builder->getInt64(0)); in EmitArrayElementAddress()
313 index[which_dimension], ir_builder->getInt64(addend), "", /*HasNUW=*/true, in BumpIndex()
Dllvm_loop.cc183 return AddLoop(suffix, start_index, end_index, ir_builder_->getInt64(1), in AddLoop()
221 return AddLoop(suffix, ir_builder_->getInt64(start_index), in AddLoop()
222 ir_builder_->getInt64(end_index), prevent_unrolling, in AddLoop()
232 return AddLoop(suffix, ir_builder_->getInt64(start_index), in AddLoop()
233 ir_builder_->getInt64(end_index), in AddLoop()
234 ir_builder_->getInt64(stride), prevent_unrolling, in AddLoop()
Dllvm_util.cc139 ? llvm::ArrayRef<llvm::Value*>({ir_builder->getInt64(0), index}) in EmitBufferIndexingGEP()
145 return EmitBufferIndexingGEP(array, ir_builder->getInt64(index), ir_builder); in EmitBufferIndexingGEP()
488 ir_builder->getInt64(tensorflow::bit_cast<int64>(&LogS64)), in EmitLogging()
490 {ir_builder->getInt64(tensorflow::bit_cast<int64>(tag)), value}); in EmitLogging()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc740 auto logical_x = builder->CreateURem(x, builder->getInt64(tile_size)); in EmitTranspose021Tiled()
743 auto logical_y = builder->CreateUDiv(x, builder->getInt64(tile_size)); in EmitTranspose021Tiled()
779 builder->CreateICmpEQ(builder->getInt64(tile_size), tile_width), in EmitTranspose021Tiled()
780 builder->CreateICmpEQ(builder->getInt64(tile_size), tile_height)), in EmitTranspose021Tiled()
784 auto source_idx = offset_dim(index, builder->getInt64(i), /*dim=*/1); in EmitTranspose021Tiled()
785 auto y_loc = builder->CreateAdd(builder->getInt64(i), logical_y); in EmitTranspose021Tiled()
796 builder->CreateAdd(tile_height, builder->getInt64(num_rows - 1)), in EmitTranspose021Tiled()
797 builder->getInt64(num_rows)), in EmitTranspose021Tiled()
798 builder->getInt64(num_rows)); in EmitTranspose021Tiled()
801 loop_name, builder->getInt64(0), tile_height_upper_bound, in EmitTranspose021Tiled()
[all …]
Dparallel_loop_emitter.cc87 ir_builder_->getInt64(launch_dimensions_.threads_per_block()), "", in EmitIndexAndSetExitBasicBlock()
103 ir_builder_->getInt64(launch_dimensions_.threads_per_block() * in EmitIndexAndSetExitBasicBlock()
110 linear_index, ir_builder_->getInt64(ShapeUtil::ElementsIn(shape_))), in EmitIndexAndSetExitBasicBlock()
Dhlo_to_ir_bindings.cc83 temp_buffer_base_, ir_builder_->getInt64(offset)); in EmitBasePointersForHlos()
119 ir_builder_->getInt64(offset)), in EmitBasePointersForHlos()
Delemental_ir_emitter.cc382 index[i], ir_builder_->getInt64(window.dimensions(i).stride())); in MakeElementGenerator()
385 ir_builder_->getInt64(window.dimensions(i).padding_low())); in MakeElementGenerator()
396 ir_builder_->getInt64(operand->shape().dimensions(i)))); in MakeElementGenerator()
/external/swiftshader/third_party/LLVM/unittests/Support/
DIRBuilderTest.cpp50 CallInst *Start3 = Builder.CreateLifetimeStart(Var3, Builder.getInt64(100)); in TEST_F()
52 EXPECT_EQ(Start1->getArgOperand(0), Builder.getInt64(-1)); in TEST_F()
53 EXPECT_EQ(Start2->getArgOperand(0), Builder.getInt64(-1)); in TEST_F()
54 EXPECT_EQ(Start3->getArgOperand(0), Builder.getInt64(100)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc574 index[i], ir_builder_.getInt64(window.dimensions(i).stride())); in HandleReduceWindow()
577 ir_builder_.getInt64(window.dimensions(i).padding_low())); in HandleReduceWindow()
585 input_index[i], ir_builder_.getInt64(ShapeUtil::GetDimension( in HandleReduceWindow()
705 source_index[i], ir_builder_.getInt64(window.dimensions(i).stride())); in HandleSelectAndScatter()
708 ir_builder_.getInt64(window.dimensions(i).padding_low())); in HandleSelectAndScatter()
711 ir_builder_.getInt64(ShapeUtil::GetDimension(operand->shape(), i))); in HandleSelectAndScatter()
967 ir_builder_.getInt64(input_batch), in HandleConvolution()
968 ir_builder_.getInt64(input_rows), in HandleConvolution()
969 ir_builder_.getInt64(input_cols), in HandleConvolution()
970 ir_builder_.getInt64(input_channels), in HandleConvolution()
[all …]
Dvector_support_library.h52 return Mul(ir_builder()->getInt64(lhs), rhs); in Mul()
65 return Add(ir_builder()->getInt64(lhs), rhs); in Add()
149 ir_builder()->getInt64(offset_elements)); in ComputeOffsetPointer()
160 return LoadVector(base_pointer, ir_builder()->getInt64(offset_elements)); in LoadVector()
171 return LoadScalar(base_pointer, ir_builder()->getInt64(offset_elements)); in LoadScalar()
183 StoreVector(value, base_pointer, ir_builder()->getInt64(offset_elements)); in StoreVector()
194 StoreScalar(base_pointer, ir_builder()->getInt64(offset_elements)); in StoreScalar()
203 return LoadBroadcast(base_pointer, ir_builder()->getInt64(offset_elements)); in LoadBroadcast()
Ddot_op_emitter.cc58 ir_builder->getInt64(matrix_size_along_minor_dim), in TileLoader()
59 ir_builder->CreateAdd(ir_builder->getInt64(i), major_dim_offset)); in TileLoader()
235 EmitOuterLoopBody(ir_builder_->getInt64(column_limit), column_remainder, in Emit()
267 llvm::Value* columns_llvm = ir_builder_->getInt64(columns); in EmitInnerLoopEpilogue()
283 ir_builder_->CreateMul(col, ir_builder_->getInt64(m_)); in EmitInnerLoopEpilogue()
458 llvm::Value* offset = ir_builder_->CreateAdd(ir_builder_->getInt64(i), row); in EmitOuterLoopBody()
476 EmitOuterLoopBody(ir_builder_->getInt64(row_limit), row_remainder); in Emit()
508 ir_builder_->CreateAdd(ir_builder_->getInt64(r), current_tile_row), in EmitInnerLoopEpilogue()
509 ir_builder_->getInt64(k_)); in EmitInnerLoopEpilogue()
991 ir_builder_->getInt64(mat_mult_dims.m), in EmitCallToRuntime()
[all …]
Dir_function.cc189 ir_builder_->getInt64(offset), AsStringRef(name))); in GetDynamicLoopBound()
213 parameter_addresses_buffer, {ir_builder->getInt64(i)}); in GetArrayFunctionCallArguments()
296 partitions[index] = ir_builder->getInt64(dim_partition.first); in EmitCallToParallelForkJoin()
298 ir_builder->getInt64(dim_partition.first + dim_partition.second); in EmitCallToParallelForkJoin()
/external/icu/icu4c/source/i18n/
Dfmtable.cpp451 Formattable::getInt64(UErrorCode& status) const in getInt64() function in Formattable
469 int64_t val = fDecimalNum->getInt64(); in getInt64()
486 getNumber().getInt64(status); in getInt64()
737 fDecimalNum->set(this->getInt64()); in internalGetCharString()
791 fValue.fInt64 = fDecimalNum->getInt64(); in adoptDigitList()
1046 return obj->getInt64(*status); in ufmt_getInt64()
Ddigitlst.h210 int64_t getInt64(void) /*const*/;
/external/llvm/unittests/IR/
DIRBuilderTest.cpp60 CallInst *Start3 = Builder.CreateLifetimeStart(Var3, Builder.getInt64(100)); in TEST_F()
62 EXPECT_EQ(Start1->getArgOperand(0), Builder.getInt64(-1)); in TEST_F()
63 EXPECT_EQ(Start2->getArgOperand(0), Builder.getInt64(-1)); in TEST_F()
64 EXPECT_EQ(Start3->getArgOperand(0), Builder.getInt64(100)); in TEST_F()
362 auto Elt1 = Builder.getInt64(-1); in TEST_F()
363 auto Elt2 = Builder.getInt64(-2); in TEST_F()
/external/llvm/lib/Transforms/Utils/
DIntegerDivision.cpp40 Shift = Builder.getInt64(63); in generateSignedRemainderCode()
112 Shift = Builder.getInt64(63); in generateSignedDivisionCode()
167 Zero = Builder.getInt64(0); in generateUnsignedDivisionCode()
168 One = Builder.getInt64(1); in generateUnsignedDivisionCode()
170 MSB = Builder.getInt64(63); in generateUnsignedDivisionCode()
/external/icu/icu4c/source/i18n/unicode/
Dfmtable.h345 int64_t getInt64(void) const { return fValue.fInt64; } in getInt64() function
362 int64_t getInt64(UErrorCode& status) const;
/external/swiftshader/third_party/LLVM/lib/VMCore/
DIRBuilder.cpp126 Size = getInt64(-1); in CreateLifetimeStart()
141 Size = getInt64(-1); in CreateLifetimeEnd()
/external/deqp/executor/
DxeTestCaseResult.cpp78 return str << value.getInt64(); in operator <<()
/external/llvm/lib/Transforms/Instrumentation/
DGCOVProfiling.cpp629 Count = Builder.CreateAdd(Count, Builder.getInt64(1)); in emitProfileArcs()
634 Builder.getInt64(Edge), in emitProfileArcs()
635 Builder.getInt64(Edge + 1)); in emitProfileArcs()
637 Counters->getValueType(), Counters, {Builder.getInt64(0), Sel}); in emitProfileArcs()
639 Count = Builder.CreateAdd(Count, Builder.getInt64(1)); in emitProfileArcs()
936 Builder.getInt64(1)); in insertIndirectCounterIncrement()
/external/icu/icu4c/source/test/intltest/
Dnumrgts.cpp2119 if (n.getInt64() != dbl) { in Test4162198()
2763 TEST_ASSERT(1000000000 == val.getInt64(status));
2774 TEST_ASSERT(100000000000000001LL == val.getInt64(status));
2783 TEST_ASSERT(100000000000000000LL == val.getInt64());
2791 TEST_ASSERT(9223372036854775807LL == val.getInt64());
2804 TEST_ASSERT((int64_t)0x8000000000000000LL == val.getInt64());
2812 TEST_ASSERT(9223372036854775807LL == val.getInt64(status));
2823 TEST_ASSERT((int64_t)0x8000000000000000LL == val.getInt64(status));
2841 TEST_ASSERT(9007199254740991LL == val.getInt64(status));
2850 TEST_ASSERT(9007199254740992LL == val.getInt64(status));
[all …]
/external/llvm/lib/IR/
DIRBuilder.cpp169 Size = getInt64(-1); in CreateLifetimeStart()
184 Size = getInt64(-1); in CreateLifetimeEnd()
318 Args.push_back(B.getInt64(ID)); in getStatepointArgs()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc1112 source_index.push_back(ir_builder_->getInt64(0)); in ElementwiseSourceIndex()
1155 /*Initializer=*/ir_builder_->getInt64(random_value()), in MakeRngElementGenerator()
1164 /*Initializer=*/ir_builder_->getInt64(graph_seed), in MakeRngElementGenerator()
1512 llvm_ir::IrArray::Index dim_index(1, ir_builder_->getInt64(i)); in MakeElementGenerator()
1556 llvm_ir::IrArray::Index dim_index(1, ir_builder_->getInt64(i)); in MakeElementGenerator()
1779 IrName(hlo, "inner"), ir_builder_->getInt64(0), in MakeElementGenerator()
1780 ir_builder_->getInt64(contracted_dim_size), in MakeElementGenerator()
1781 ir_builder_->getInt64(1), ir_builder_); in MakeElementGenerator()

123