| /external/tensorflow/tensorflow/stream_executor/ |
| fft.h | 111 ScratchAllocator *scratch_allocator) = 0;
116 ScratchAllocator *scratch_allocator) = 0;
121 bool in_place_fft, ScratchAllocator *scratch_allocator) = 0;
168 ScratchAllocator *scratch_allocator) = 0;
177 Stream *stream, Plan *plan, ScratchAllocator *scratch_allocator) = 0;
226 ScratchAllocator *scratch_allocator) override; \
229 bool in_place_fft, ScratchAllocator *scratch_allocator) override; \
232 fft::Type type, bool in_place_fft, ScratchAllocator *scratch_allocator) \
243 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) \
246 ScratchAllocator *scratch_allocator) \
[all...] |
| dnn.h | [all...] |
| stream.h | 316 ScratchAllocator *scratch_allocator);
329 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator);
341 DeviceMemory<float> *output, ScratchAllocator *scratch_allocator);
350 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator);
359 DeviceMemory<float> *output, ScratchAllocator *scratch_allocator);
368 DeviceMemory<float> *output, ScratchAllocator *scratch_allocator,
379 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator,
393 DeviceMemory<double> *output, ScratchAllocator *scratch_allocator,
407 DeviceMemory<float> *output, ScratchAllocator *scratch_allocator,
422 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator,
[all...] |
| stream.cc | 425 ScratchAllocator *scratch_allocator) {
439 activation_mode, output_descriptor, output, scratch_allocator,
459 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator) {
473 activation_mode, output_descriptor, output, scratch_allocator,
492 ScratchAllocator *scratch_allocator) {
506 activation_mode, output_descriptor, output, scratch_allocator,
522 DeviceMemory<Eigen::half> *output, ScratchAllocator *scratch_allocator) {
532 convolution_descriptor, output_descriptor, output, scratch_allocator,
549 ScratchAllocator *scratch_allocator) {
559 convolution_descriptor, output_descriptor, output, scratch_allocator,
[all...] |
| blas.h | [all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_fft.h | 27 #include "tensorflow/stream_executor/scratch_allocator.h"
73 int batch_count, ScratchAllocator *scratch_allocator);
78 ScratchAllocator *scratch_allocator);
81 ScratchAllocator *scratch_allocator);
|
| cuda_fft.cc | 115 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) {
134 if (scratch_allocator == nullptr) {
225 return UpdateScratchAllocator(stream, scratch_allocator);
229 if (scratch_allocator == nullptr) {
266 return UpdateScratchAllocator(stream, scratch_allocator);
275 ScratchAllocator *scratch_allocator) {
280 /*output_distance=*/0, type, 1, scratch_allocator);
284 Stream *stream, ScratchAllocator *scratch_allocator) {
287 scratch_allocator->AllocateBytes(stream, scratch_size_bytes_);
332 parent_, stream, 1, elem_count, type, /*scratch_allocator=*/nullptr)
[all...] |
| cuda_dnn.h | 252 ScratchAllocator* scratch_allocator,
271 ScratchAllocator* scratch_allocator,
285 DeviceMemory<double>* output_data, ScratchAllocator* scratch_allocator,
299 DeviceMemory<float>* output_data, ScratchAllocator* scratch_allocator,
317 ScratchAllocator* scratch_allocator,
331 DeviceMemory<int8>* output_data, ScratchAllocator* scratch_allocator,
382 ScratchAllocator* scratch_allocator,
394 ScratchAllocator* scratch_allocator,
406 ScratchAllocator* scratch_allocator,
418 ScratchAllocator* scratch_allocator,
[all...] |
| cuda_dnn.cc | 39 #include "tensorflow/stream_executor/scratch_allocator.h"
[all...] |
| cuda_blas.h | 117 int batch_count, ScratchAllocator *scratch_allocator);
|
| cuda_blas.cc | 56 #include "tensorflow/stream_executor/scratch_allocator.h"
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_convolution_runner.cc | 78 se::ScratchAllocator* scratch_allocator, const Window& window,
178 scratch_allocator, algorithm, profile_result);
184 scratch_allocator, algorithm, profile_result);
190 scratch_allocator, algorithm, profile_result);
226 ScratchBufAllocator scratch_allocator(scratch_buf);
229 &scratch_allocator, window, dnums, algorithm,
238 perftools::gputools::ScratchAllocator* scratch_allocator,
250 se::DeviceMemory<float>(output_buf), scratch_allocator, window, dnums,
257 scratch_allocator, window, dnums, algorithm,
|
| cudnn_convolution_runner.h | 89 perftools::gputools::ScratchAllocator* scratch_allocator,
|
| convolution_thunk.h | 84 perftools::gputools::Stream* stream, ScratchAllocator* scratch_allocator,
|
| fft_thunk.cc | 133 FftScratchAllocator scratch_allocator(buffer_allocations.device_ordinal(),
165 fft_type_, kInPlaceFft, batch_size, &scratch_allocator);
169 stream, fft_plan_.get(), &scratch_allocator);
|
| cudnn_convolution_algorithm_picker.cc | 219 ScratchAllocator scratch_allocator(device_ordinal, allocator);
227 output_buf.ValueOrDie(), &scratch_allocator, window,
232 int64 scratch_bytes_used = scratch_allocator.TotalAllocatedBytes();
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.cc | 553 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
562 &scratch_allocator, dnn::AlgorithmConfig(profile_algorithm),
571 if (scratch_allocator.TotalByteSize() == 0 &&
593 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
600 &output_ptr, &scratch_allocator, algorithm_config,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_ops_3d.cc | 405 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
411 output_desc, &output_ptr, &scratch_allocator,
420 if (scratch_allocator.TotalByteSize() == 0 &&
441 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
446 &output_ptr, &scratch_allocator,
|
| conv_grad_ops_3d.cc | [all...] |
| conv_ops.cc | 718 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
724 output_desc, &output_ptr, &scratch_allocator,
733 if (scratch_allocator.TotalByteSize() == 0 &&
756 CudnnScratchAllocator scratch_allocator(ConvolveScratchSize, ctx);
761 &output_ptr, &scratch_allocator,
[all...] |
| conv_grad_filter_ops.cc | [all...] |
| conv_grad_input_ops.cc | [all...] |
| batch_matmul_op_impl.h | 385 CublasScratchAllocator scratch_allocator(context);
392 batch_size, &scratch_allocator)
|
| fft_ops.cc | 401 CufftScratchAllocator scratch_allocator(CufftScratchSize, ctx);
406 kFftType, kInPlaceFft, batch_size, &scratch_allocator);
|
| /system/update_engine/payload_generator/ |
| inplace_generator.cc | 142 DummyExtentAllocator scratch_allocator; local
154 scratch_allocator.Allocate(graph_utils::EdgeWeight(*graph, edge));
|
