Searched refs:allocation_ (Results 1 – 6 of 6) sorted by relevance
| /external/v8/src/compiler/ |
| D | allocation-builder.h | 23 allocation_(nullptr), in AllocationBuilder()
33 allocation_ =
36 effect_ = allocation_;
41 effect_ = graph()->NewNode(simplified()->StoreField(access), allocation_, in Store()
47 effect_ = graph()->NewNode(simplified()->StoreElement(access), allocation_, in Store()
84 NodeProperties::SetType(allocation_, NodeProperties::GetType(node)); in FinishAndChange()
85 node->ReplaceInput(0, allocation_); in FinishAndChange()
92 return graph()->NewNode(common()->FinishRegion(), allocation_, effect_); in Finish()
104 Node* allocation_; variable
|
| /external/tensorflow/tensorflow/lite/ |
| D | model.h | 133 const Allocation* allocation() const { return allocation_.get(); } in allocation()
160 std::unique_ptr<Allocation> allocation_; variable
216 const Allocation* allocation_ = nullptr; variable
|
| D | model.cc | 167 if (!tflite::ModelBufferHasIdentifier(allocation_->base())) { in CheckModelIdentifier()
168 const char* ident = flatbuffers::GetBufferIdentifier(allocation_->base()); in CheckModelIdentifier()
184 allocation_(std::move(allocation)) { in FlatBufferModel()
185 if (!allocation_->valid() || !CheckModelIdentifier()) return; in FlatBufferModel()
187 model_ = ::tflite::GetModel(allocation_->base()); in FlatBufferModel()
197 allocation_(model.allocation()) {} in InterpreterBuilder()
422 buffer_size, allocation_) != kTfLiteOk) { in ParseTensors()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 136 allocation_(allocation), in XRTBufferAllocation()
141 GetAllocStats()->ReportAlloc(device_ordinal_, allocation_.size()); in XRTBufferAllocation()
149 GetAllocStats()->ReportFree(device_ordinal_, allocation_.size()); in ~XRTBufferAllocation()
152 Status s = allocator_->Deallocate(device_ordinal_, allocation_); in ~XRTBufferAllocation()
155 VLOG(2) << "Freed buffer at " << allocation_.opaque(); in ~XRTBufferAllocation()
159 return allocation_; in allocation()
164 allocation_ = se::DeviceMemoryBase(); in DiscardAllocation()
|
| D | xrt_state.h | 70 se::DeviceMemoryBase allocation_; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.h | 166 : allocation_(allocation), offset_(offset), size_(size) {} in Slice()
168 const BufferAllocation* allocation() const { return allocation_; } in allocation()
169 Index index() const { return allocation_->index(); } in index()
201 const BufferAllocation* allocation_ = nullptr;
|