Searched refs:base_ptr_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_cpurt.h | 41 : TensorBuffer(data), base_ptr_(base_ptr), size_(size), owner_(owner) {} in MemrefTensorBuffer()
44 if (owner_) free(base_ptr_); in ~MemrefTensorBuffer()
57 void* base_ptr_;
|
| D | tf_jitrt.h | 50 : TensorBuffer(data), base_ptr_(base_ptr), size_(size), owner_(owner) {} in MemrefTensorBuffer()
53 if (owner_) free(base_ptr_); in ~MemrefTensorBuffer()
66 void* base_ptr_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 185 : base_ptr_(base_ptr), in IrArray()
189 CHECK(base_ptr_->getType()->isPointerTy()); in IrArray()
190 CHECK(llvm::cast<llvm::PointerType>(base_ptr_->getType()) in IrArray()
478 return base_ptr_; in EmitArrayElementAddress()
489 type, b->CreateBitCast(base_ptr_, type->getPointerTo()), index.linear(), in EmitArrayElementAddress()
515 return b->CreateInBoundsGEP(pointee_type_, base_ptr_, gep_indices, in EmitArrayElementAddress()
557 b->CreatePointerCast(base_ptr_, new_ir_type->getPointerTo()), new_ir_type, in CastToShape()
|
| D | ir_array.h | 220 IrArray() : base_ptr_(nullptr) {} in IrArray()
233 llvm::Value* GetBasePointer() const { return base_ptr_; } in GetBasePointer()
328 llvm::Value* base_ptr_; variable
|