Searched refs:InBoundsGEP (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 895 InBoundsGEP(selected_index_address, {b_.getInt32(i)}); in HandleSelectAndScatter()
946 InBoundsGEP(selected_index_address, {b_.getInt32(i)}); in HandleSelectAndScatter()
1673 loc = InBoundsGEP(kernel_args.at(slice.allocation()), in BuildKernelThunk()
1683 loc = Load(InBoundsGEP(loc, {b_.getInt64(idx)})); in BuildKernelThunk()
2693 Store(init_ir_value, InBoundsGEP(partial_result_address, {b_.getInt32(i)})); in EmitPrologueForOneReduction()
2818 InBoundsGEP(reduction_info->GetCurrentOutputLinearIndexAddress(), in EmitEpilogueForReduction()
2841 InBoundsGEP(partial_result_addresses[i], {b_.getInt32(j)})}, in EmitEpilogueForReduction()
2846 InBoundsGEP(partial_result_addresses[i], {b_.getInt32(j)}))); in EmitEpilogueForReduction()
2871 InBoundsGEP(reduction_info->GetCurrentOutputLinearIndexAddress(), in EmitTileElementForReduction()
2932 InBoundsGEP(partial_reduction_result_addresses[i], in EmitTileElementForReduction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_builder_mixin.h | 135 llvm::Value* InBoundsGEP(llvm::Value* ptr,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 852 InBoundsGEP(selected_index_address, {b_.getInt32(i)}); in HandleSelectAndScatter()
895 InBoundsGEP(selected_index_address, {b_.getInt32(i)}); in HandleSelectAndScatter()
2240 InBoundsGEP(operands_alloca, {b_.getInt64(i)}); in HandleCustomCall()
2913 InBoundsGEP(tempbuf_address_base, b_.getInt64(slice.offset())); in EmitGlobalBufferPointer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 1651 InBoundsGEP(sample_address, b_->getInt32(idx))); in MakePhiloxRngElementGenerator()
1659 InBoundsGEP(PointerCast(sample_address, raw_value_ty->getPointerTo()), in MakePhiloxRngElementGenerator()
|
| /external/swiftshader/third_party/llvm-7.0/llvm/lib/Transforms/Scalar/ |
| D | SROA.cpp | 3321 Value *InBoundsGEP = in emitFunc() local
3323 StoreInst *Store = IRB.CreateStore(ExtractValue, InBoundsGEP); in emitFunc()
|
| /external/llvm/lib/Transforms/Scalar/ |
| D | SROA.cpp | 3115 Value *InBoundsGEP = in emitFunc() local
3117 Value *Store = IRB.CreateStore(ExtractValue, InBoundsGEP); in emitFunc()
|