Home
last modified time | relevance | path

Searched refs:buffer_slice (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.cc37 BufferAllocation::Slice buffer_slice; in AddAliasingInformationToIrArray() local
42 buffer_slice = BufferAllocation::Slice(kParameterAllocation, 0, 0); in AddAliasingInformationToIrArray()
51 buffer_slice = unique_slice.ValueOrDie(); in AddAliasingInformationToIrArray()
55 llvm::MDNode*& alias_scope_md = alias_scope_metadata_[buffer_slice]; in AddAliasingInformationToIrArray()
58 GetAliasScopeMetadataForBuffer(buffer_slice, GetAliasDomain()); in AddAliasingInformationToIrArray()
66 llvm::MDNode*& noalias_md = noalias_metadata_[{buffer_slice, &hlo}]; in AddAliasingInformationToIrArray()
68 noalias_md = GetNoaliasMetadataForBuffer(buffer_slice, GetAliasDomain(), in AddAliasingInformationToIrArray()
106 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain) { in GetAliasScopeMetadataForBuffer() argument
109 if (buffer_slice.allocation() == kParameterAllocation) { in GetAliasScopeMetadataForBuffer()
115 "buffer: " + buffer_slice.ToString(), domain); in GetAliasScopeMetadataForBuffer()
[all …]
Dalias_analysis.h49 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain);
60 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_allocations.cc75 const BufferAllocation::Slice& buffer_slice) const { in GetDeviceAddress()
76 se::DeviceMemoryBase base = GetDeviceAddress(buffer_slice.index()); in GetDeviceAddress()
77 CHECK_LE(buffer_slice.offset(), base.size()); in GetDeviceAddress()
78 CHECK_LE(buffer_slice.offset() + buffer_slice.size(), base.size()); in GetDeviceAddress()
80 static_cast<char*>(base.opaque()) + buffer_slice.offset(), in GetDeviceAddress()
81 buffer_slice.size()); in GetDeviceAddress()
Dbuffer_allocations.h68 const BufferAllocation::Slice& buffer_slice) const;
Dir_emitter_unnested.cc1750 TF_ASSIGN_OR_RETURN(slice.buffer_slice, in GetMlirEmitterInput()
3320 hlo_buffer_slice.buffer_slice = maybe_slice->first; in GetHloBufferSlices()
3353 buffers_needed.insert(slice->buffer_slice.allocation()); in BuildKernelThunkFromBufferSlices()
3416 const BufferAllocation::Slice& buffer_slice = slice->buffer_slice; in BuildKernelThunkFromBufferSlices() local
3420 if (buffer_slice.allocation()->is_constant()) { in BuildKernelThunkFromBufferSlices()
3423 *buffer_slice.allocation())); in BuildKernelThunkFromBufferSlices()
3426 loc = InBoundsGEP(kernel_args.at(buffer_slice.allocation()), in BuildKernelThunkFromBufferSlices()
3427 {b_.getInt64(buffer_slice.offset())}); in BuildKernelThunkFromBufferSlices()
3476 << hlo_buffer_slice->buffer_slice.ToString() << " at GTE index " in BuildKernelThunk()
3493 buffers_written.insert(slice.buffer_slice); in BuildKernelThunkForMlirImpl()
[all …]
Dir_emitter_unnested.h35 BufferAllocation::Slice buffer_slice; member
/external/python/cpython2/Objects/
Dbufferobject.c485 buffer_slice(PyBufferObject *self, Py_ssize_t left, Py_ssize_t right) in buffer_slice() function
839 (ssizessizeargfunc)buffer_slice, /*sq_slice*/