HomeSort by relevance Sort by last modified time
    Searched refs:BufferAllocation (Results 1 - 25 of 49) sorted by null

1 2

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_batchnorm_thunk.h 49 CudnnBatchNormForwardInferenceThunk(const BufferAllocation::Slice& operand,
50 const BufferAllocation::Slice& scale,
51 const BufferAllocation::Slice& offset,
52 const BufferAllocation::Slice& mean,
53 const BufferAllocation::Slice& variance,
55 const BufferAllocation::Slice& output,
68 BufferAllocation::Slice operand_;
69 BufferAllocation::Slice scale_;
70 BufferAllocation::Slice offset_;
71 BufferAllocation::Slice mean_
    [all...]
convolution_thunk.h 47 std::vector<BufferAllocation::Slice> operand_slices,
48 BufferAllocation::Slice result_slice,
49 BufferAllocation::Slice scratch_slice,
50 BufferAllocation::Slice tuple_result_slice);
62 std::vector<BufferAllocation::Slice> operand_buffers_;
63 BufferAllocation::Slice result_buffer_;
64 BufferAllocation::Slice scratch_buffer_;
65 BufferAllocation::Slice tuple_result_buffer_;
nccl_all_reduce_thunk.h 44 const BufferAllocation::Slice& source_buffer,
45 const BufferAllocation::Slice& destination_buffer,
55 const BufferAllocation::Slice source_buffer_;
56 const BufferAllocation::Slice destination_buffer_;
cholesky_thunk.h 45 BufferAllocation::Slice a_buffer,
46 BufferAllocation::Slice workspace_buffer,
47 BufferAllocation::Slice info_buffer,
61 const BufferAllocation::Slice a_buffer_;
62 const BufferAllocation::Slice workspace_buffer_;
63 const BufferAllocation::Slice info_buffer_;
copy_thunk.h 37 const BufferAllocation::Slice& destination_buffer,
49 const BufferAllocation::Slice destination_buffer_;
59 DeviceToDeviceCopyThunk(const BufferAllocation::Slice& source_buffer,
60 const BufferAllocation::Slice& destination_buffer,
72 const BufferAllocation::Slice source_buffer_;
73 const BufferAllocation::Slice destination_buffer_;
gemm_thunk.h 40 GemmThunk(const BufferAllocation::Slice& lhs_buffer,
41 const BufferAllocation::Slice& rhs_buffer,
42 const BufferAllocation::Slice& output_buffer,
65 const BufferAllocation::Slice lhs_buffer_;
66 const BufferAllocation::Slice rhs_buffer_;
67 const BufferAllocation::Slice output_buffer_;
tuple_thunk.h 37 TupleThunk(absl::Span<const BufferAllocation::Slice> tuple_element_buffers,
38 const BufferAllocation::Slice& dest_buffer,
53 const std::vector<BufferAllocation::Slice> tuple_element_buffers_;
54 const BufferAllocation::Slice dest_buffer_;
infeed_thunk.h 37 InfeedThunk(const ShapeTree<BufferAllocation::Slice>& infeed_slices,
48 const ShapeTree<BufferAllocation::Slice> infeed_slices_;
outfeed_thunk.h 35 OutfeedThunk(ShapeTree<BufferAllocation::Slice> outfeed_slices,
46 const ShapeTree<BufferAllocation::Slice> outfeed_slices_;
conditional_thunk.h 46 const BufferAllocation::Slice& branch_index_buffer_index,
47 absl::Span<const BufferAllocation::Slice> branch_operand_buffer_indexes,
62 BufferAllocation::Slice branch_index_buffer_index_;
63 std::vector<BufferAllocation::Slice> branch_operand_buffer_indexes_;
memset_thunk.h 35 explicit MemzeroThunk(const BufferAllocation::Slice& dest,
44 const BufferAllocation::Slice dest_;
52 const BufferAllocation::Slice& dest,
62 const BufferAllocation::Slice dest_;
triangular_solve_thunk.h 42 const BufferAllocation::Slice& a_buffer,
43 const BufferAllocation::Slice& b_buffer,
61 const BufferAllocation::Slice a_buffer_;
62 const BufferAllocation::Slice b_buffer_;
buffer_allocations.h 44 void RegisterBuffer(BufferAllocation::Index index,
56 absl::flat_hash_map<BufferAllocation::Index, se::DeviceMemoryBase>
72 BufferAllocation::Index buffer_index) const;
77 const BufferAllocation::Slice& buffer_slice) const;
86 BufferAllocations(BufferAllocation::Index buffer_count, int device_ordinal,
95 void SetBuffer(BufferAllocation::Index buffer_index,
while_thunk.h 42 WhileThunk(const BufferAllocation::Slice& condition_result_buffer_index,
56 const BufferAllocation::Slice condition_result_buffer_index_;
cudnn_batchnorm_thunk.cc 79 const BufferAllocation::Slice& operand,
80 const BufferAllocation::Slice& scale, const BufferAllocation::Slice& offset,
81 const BufferAllocation::Slice& mean,
82 const BufferAllocation::Slice& variance, float epsilon, int64 feature_index,
83 const BufferAllocation::Slice& output, const HloInstruction* hlo)
136 const BufferAllocation::Slice& operand,
137 const BufferAllocation::Slice& scale, const BufferAllocation::Slice& offset,
139 const BufferAllocation::Slice& output_data
    [all...]
convolution_thunk.cc 34 std::vector<BufferAllocation::Slice> operand_slices,
35 BufferAllocation::Slice result_slice, BufferAllocation::Slice scratch_slice,
36 BufferAllocation::Slice tuple_result_slice)
fft_thunk.h 66 const BufferAllocation::Slice& input_buffer,
67 const BufferAllocation::Slice& output_buffer,
87 const BufferAllocation::Slice input_buffer_;
88 const BufferAllocation::Slice output_buffer_;
buffer_allocations.cc 35 void BufferAllocations::Builder::RegisterBuffer(BufferAllocation::Index index,
47 for (BufferAllocation::Index i = 0; i < num_buffers; ++i) {
48 const BufferAllocation& allocation = buffer_assignment->GetAllocation(i);
109 for (BufferAllocation::Index i = 0; i < num_buffers; ++i) {
134 for (BufferAllocation::Index i = 0; i < num_buffers; ++i) {
135 const BufferAllocation& allocation = buffer_assignment_->GetAllocation(i);
154 BufferAllocation::Index buffer_index) const {
161 const BufferAllocation::Slice& buffer_slice) const {
170 void BufferAllocations::SetBuffer(BufferAllocation::Index buffer_index,
kernel_thunk.h 50 KernelThunk(absl::Span<const BufferAllocation* const> args,
71 const std::vector<const BufferAllocation*> args_;
copy_thunk.cc 26 const BufferAllocation::Slice& destination_buffer, uint64 mem_size,
44 const BufferAllocation::Slice& source_buffer,
45 const BufferAllocation::Slice& destination_buffer, uint64 mem_size,
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
buffer_assignment_util.h 30 string ConstantBufferAllocationToGlobalName(const BufferAllocation& allocation);
34 const Literal& LiteralForConstantAllocation(const BufferAllocation& allocation);
alias_analysis.h 49 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain);
60 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain,
79 absl::flat_hash_map<BufferAllocation::Slice, llvm::MDNode*>
84 absl::flat_hash_map<BufferAllocation::Slice, llvm::MDNode*> noalias_metadata_;
alias_analysis.cc 30 static const BufferAllocation* kParameterAllocation = new BufferAllocation(
36 BufferAllocation::Slice buffer_slice;
41 buffer_slice = BufferAllocation::Slice(kParameterAllocation, 0, 0);
43 const std::set<BufferAllocation::Slice> slices =
109 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain) {
124 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain,
165 std::set<BufferAllocation::Slice> buffers;
172 const BufferAllocation::Slice noalias_slice =
198 for (const BufferAllocation::Slice noalias_slice : buffers)
    [all...]
buffer_assignment_util.cc 22 const BufferAllocation& allocation) {
33 // BufferAllocation as the LogicalBuffer for the (init) constant.
58 const BufferAllocation& allocation) {
68 const BufferAllocation& allocation) {
  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_assignment.h 57 // of the allocation, represented by a Slice. A single BufferAllocation may hold
59 // single BufferAllocation may also hold LogicalBuffers with overlapping
65 class BufferAllocation {
71 BufferAllocation(Index index, int64 size, LogicalBuffer::Color color)
73 ~BufferAllocation() {}
165 Slice(const BufferAllocation* allocation, int64 offset, int64 size)
168 const BufferAllocation* allocation() const { return allocation_; }
201 const BufferAllocation* allocation_ = nullptr;
234 // allocation. A single BufferAllocation may include multiple heap traces
262 bool operator==(const BufferAllocation& other) const
    [all...]

Completed in 835 milliseconds

1 2