| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_thunk.h | 52 const BufferAllocation::Slice& operand,
53 const BufferAllocation::Slice& scale,
54 const BufferAllocation::Slice& offset,
55 const BufferAllocation::Slice& mean,
56 const BufferAllocation::Slice& variance,
57 const BufferAllocation::Slice& output);
68 BufferAllocation::Slice operand_;
69 BufferAllocation::Slice scale_;
70 BufferAllocation::Slice offset_;
71 BufferAllocation::Slice mean_;
[all …]
|
| D | cudnn_batchnorm_thunk.cc | 36 const BufferAllocation::Slice& operand, in CudnnBatchNormForwardInferenceThunk()
37 const BufferAllocation::Slice& scale, const BufferAllocation::Slice& offset, in CudnnBatchNormForwardInferenceThunk()
38 const BufferAllocation::Slice& mean, in CudnnBatchNormForwardInferenceThunk()
39 const BufferAllocation::Slice& variance, in CudnnBatchNormForwardInferenceThunk()
40 const BufferAllocation::Slice& output) in CudnnBatchNormForwardInferenceThunk()
75 const BufferAllocation::Slice& operand, in CudnnBatchNormForwardTrainingThunk()
76 const BufferAllocation::Slice& scale, const BufferAllocation::Slice& offset, in CudnnBatchNormForwardTrainingThunk()
77 const BufferAllocation::Slice& output_data, in CudnnBatchNormForwardTrainingThunk()
78 const BufferAllocation::Slice& output_mean, in CudnnBatchNormForwardTrainingThunk()
79 const BufferAllocation::Slice& output_inv_stddev) in CudnnBatchNormForwardTrainingThunk()
[all …]
|
| D | convolution_thunk.h | 46 std::vector<BufferAllocation::Slice> operand_slices,
47 BufferAllocation::Slice result_slice,
48 BufferAllocation::Slice scratch_slice);
56 std::vector<BufferAllocation::Slice> operand_buffers_;
57 BufferAllocation::Slice result_buffer_;
58 BufferAllocation::Slice scratch_buffer_;
|
| D | copy_thunk.h | 37 const BufferAllocation::Slice& destination_buffer,
47 const BufferAllocation::Slice destination_buffer_;
58 const BufferAllocation::Slice& source_buffer,
59 const BufferAllocation::Slice& destination_buffer,
68 const BufferAllocation::Slice source_buffer_;
69 const BufferAllocation::Slice destination_buffer_;
|
| D | cholesky_thunk.h | 45 BufferAllocation::Slice a_buffer,
46 BufferAllocation::Slice workspace_buffer,
47 BufferAllocation::Slice info_buffer, PrimitiveType type,
58 const BufferAllocation::Slice a_buffer_;
59 const BufferAllocation::Slice workspace_buffer_;
60 const BufferAllocation::Slice info_buffer_;
|
| D | gemm_thunk.h | 52 const BufferAllocation::Slice& lhs_buffer,
53 const BufferAllocation::Slice& rhs_buffer,
54 const BufferAllocation::Slice& output_buffer,
64 const BufferAllocation::Slice lhs_buffer_;
65 const BufferAllocation::Slice rhs_buffer_;
66 const BufferAllocation::Slice output_buffer_;
|
| D | buffer_allocations.cc | 37 absl::Span<const BufferAllocation> allocations) { in TearDown()
42 for (BufferAllocation::Index i = 0; i < num_buffers; ++i) { in TearDown()
43 const BufferAllocation& allocation = allocations[i]; in TearDown()
61 BufferAllocation::Index buffer_index) const { in GetDeviceAddress()
68 BufferAllocation::Index buffer_index) { in GetMutableDeviceAddress()
75 const BufferAllocation::Slice& buffer_slice) const { in GetDeviceAddress()
|
| D | buffer_allocations.h | 59 BufferAllocation::Index buffer_index) const;
63 BufferAllocation::Index buffer_index);
68 const BufferAllocation::Slice& buffer_slice) const;
73 absl::Span<const BufferAllocation> allocations);
77 for (BufferAllocation::Index i = 0; i < buffers_.size(); ++i) { in ToString()
|
| D | collective_permute_thunk.h | 33 const BufferAllocation::Slice& src, const BufferAllocation::Slice& dest);
39 const BufferAllocation::Slice src_;
40 const BufferAllocation::Slice dest_;
|
| D | custom_call_thunk.h | 42 std::vector<BufferAllocation::Slice> operands,
43 std::vector<BufferAllocation::Slice> results,
50 const std::vector<BufferAllocation::Slice> operands_;
51 const std::vector<BufferAllocation::Slice> results_;
|
| D | memset_thunk.h | 36 const BufferAllocation::Slice& dest) in MemzeroThunk()
42 const BufferAllocation::Slice dest_;
50 const BufferAllocation::Slice& dest) in Memset32BitValueThunk()
59 const BufferAllocation::Slice dest_;
|
| D | replica_id_thunk.h | 32 const BufferAllocation::Slice& dest) in ReplicaOrPartitionIdThunk()
36 const BufferAllocation::Slice dest_;
41 ReplicaIdThunk(ThunkInfo thunk_info, const BufferAllocation::Slice& dest) in ReplicaIdThunk()
47 PartitionIdThunk(ThunkInfo thunk_info, const BufferAllocation::Slice& dest) in PartitionIdThunk()
|
| D | triangular_solve_thunk.h | 43 const BufferAllocation::Slice& a_buffer,
44 const BufferAllocation::Slice& b_buffer,
59 const BufferAllocation::Slice a_buffer_;
60 const BufferAllocation::Slice b_buffer_;
|
| D | conditional_thunk.h | 59 const BufferAllocation::Slice& branch_index_buffer_index,
60 absl::Span<const BufferAllocation::Slice> branch_operand_buffer_indexes);
71 BufferAllocation::Slice branch_index_buffer_index_;
72 std::vector<BufferAllocation::Slice> branch_operand_buffer_indexes_;
|
| D | custom_call_thunk.cc | 31 std::vector<BufferAllocation::Slice> operands, in CustomCallThunk()
32 std::vector<BufferAllocation::Slice> results, in CustomCallThunk()
44 for (const std::vector<BufferAllocation::Slice>& slices : in ExecuteOnStream()
46 for (const BufferAllocation::Slice& slice : slices) { in ExecuteOnStream()
|
| D | fft_thunk.h | 68 const BufferAllocation::Slice& input_buffer,
69 const BufferAllocation::Slice& output_buffer,
93 const BufferAllocation::Slice input_buffer_;
94 const BufferAllocation::Slice output_buffer_;
|
| D | gpu_executable.h | 79 std::vector<BufferAllocation> allocations;
130 absl::Span<const BufferAllocation> GetAllocations() const { in GetAllocations()
146 absl::flat_hash_map<BufferAllocation::Index, se::DeviceMemoryBase>;
168 const BufferAllocation& allocation,
202 const std::vector<BufferAllocation> allocations_;
|
| D | convolution_thunk.cc | 35 std::vector<BufferAllocation::Slice> operand_slices, in ConvolutionThunk()
36 BufferAllocation::Slice result_slice, BufferAllocation::Slice scratch_slice) in ConvolutionThunk()
|
| D | copy_thunk.cc | 26 const BufferAllocation::Slice& destination_buffer, uint64 mem_size) in HostToDeviceCopyThunk()
42 ThunkInfo thunk_info, const BufferAllocation::Slice& source_buffer, in DeviceToDeviceCopyThunk()
43 const BufferAllocation::Slice& destination_buffer, uint64 mem_size) in DeviceToDeviceCopyThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.h | 68 class BufferAllocation {
74 BufferAllocation(Index index, int64 size, LogicalBuffer::Color color) in BufferAllocation() function
76 ~BufferAllocation() {} in ~BufferAllocation()
170 Slice(const BufferAllocation* allocation, int64 offset, int64 size) in Slice()
173 const BufferAllocation* allocation() const { return allocation_; } in allocation()
206 const BufferAllocation* allocation_ = nullptr;
267 bool operator==(const BufferAllocation& other) const {
270 bool operator!=(const BufferAllocation& other) const {
273 bool operator<(const BufferAllocation& other) const {
350 std::ostream& operator<<(std::ostream& out, const BufferAllocation& s);
[all …]
|
| D | buffer_assignment.cc | 228 string BufferAllocation::Slice::ToString() const { in ToString()
233 BufferAllocation::Slice BufferAllocation::GetSlice( in GetSlice()
239 void BufferAllocation::AddAssignment(const HloValue& buffer, int64 offset, in AddAssignment()
270 BufferAllocationProto BufferAllocation::ToProto() const { in ToProto()
307 const BufferAllocation& alloc) { in GetEntryParameterInstruction()
322 const BufferAllocation& alloc) { in GetOutputInstruction()
337 string BufferAllocation::ToString() const { in ToString()
384 std::ostream& operator<<(std::ostream& out, const BufferAllocation& buffer) { in operator <<()
389 std::ostream& operator<<(std::ostream& out, const BufferAllocation::Slice& s) { in operator <<()
402 const BufferAllocation& BufferAssignment::GetAssignedAllocation( in GetAssignedAllocation()
[all …]
|
| D | buffer_assignment_test.cc | 249 const BufferAllocation& GetAssignedInputAllocation( in GetAssignedInputAllocation()
252 const BufferAllocation& buffer = in GetAssignedInputAllocation()
260 const BufferAllocation& GetAssignedOutputAllocation( in GetAssignedOutputAllocation()
263 const BufferAllocation& buffer = GetTopLevelAllocation(buffers, hlo); in GetAssignedOutputAllocation()
268 const BufferAllocation& GetAllocation(const BufferAssignment& buffers, in GetAllocation()
273 const BufferAllocation& GetTopLevelAllocation(const BufferAssignment& buffers, in GetTopLevelAllocation()
321 absl::flat_hash_set<BufferAllocation::Slice> a_slices; in BuffersDistinct()
456 BufferAllocation paramscalar_buffer = in TEST_F()
458 BufferAllocation param0_buffer = GetAssignedInputAllocation(*buffers, param0); in TEST_F()
459 BufferAllocation param1_buffer = GetAssignedInputAllocation(*buffers, param1); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | alias_analysis.cc | 31 static const BufferAllocation* kParameterAllocation = new BufferAllocation(
37 BufferAllocation::Slice buffer_slice; in AddAliasingInformationToIrArray()
42 buffer_slice = BufferAllocation::Slice(kParameterAllocation, 0, 0); in AddAliasingInformationToIrArray()
106 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain) { in GetAliasScopeMetadataForBuffer()
121 const BufferAllocation::Slice& buffer_slice, llvm::MDNode* domain, in GetNoaliasMetadataForBuffer()
173 std::set<BufferAllocation::Slice> buffers; in GetNoaliasMetadataForBuffer()
175 const BufferAllocation::Slice noalias_slice = in GetNoaliasMetadataForBuffer()
201 for (const BufferAllocation::Slice noalias_slice : buffers) { in GetNoaliasMetadataForBuffer()
|
| D | 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<std::pair<BufferAllocation::Slice, const HloInstruction*>,
|
| D | buffer_assignment_util.cc | 22 const BufferAllocation& allocation) { in InstrForConstantBufferAllocation()
73 const BufferAllocation& allocation) { in ConstantBufferAllocationToGlobalName()
79 const BufferAllocation& allocation) { in LiteralForConstantAllocation()
|