Home
last modified time | relevance | path

Searched refs:ScratchSpace (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.h137 class ScratchSpace;
171 ScratchSpace<Scalar> GetScratchSpace(const TensorShape& shape,
175 ScratchSpace<Scalar> GetScratchSpace(int64 size,
377 class ScratchSpace {
379 ScratchSpace(OpKernelContext* context, int64 size, bool on_host)
380 : ScratchSpace(context, TensorShape({size}), "", on_host) {}
382 ScratchSpace(OpKernelContext* context, int64 size,
384 : ScratchSpace(context, TensorShape({size}), debug_info, on_host) {}
386 ScratchSpace(OpKernelContext* context, const TensorShape& shape,
400 virtual ~ScratchSpace() {}
[all …]
Drocm_solvers.h65 class ScratchSpace;
100 class ScratchSpace {
102 ScratchSpace(OpKernelContext* context, int64 size, bool on_host)
103 : ScratchSpace(context, TensorShape({size}), "", on_host) {}
105 ScratchSpace(OpKernelContext* context, int64 size, const string& debug_info,
107 : ScratchSpace(context, TensorShape({size}), debug_info, on_host) {}
109 ScratchSpace(OpKernelContext* context, const TensorShape& shape,
123 virtual ~ScratchSpace() {}
Dcuda_solvers.cc420 ScratchSpace<uint8> dev_a_dev_ptrs = in PotrfBatchedImpl()
766 ScratchSpace<uint8> dev_a_dev_ptrs = in GetrfBatchedImpl()
801 ScratchSpace<uint8> dev_a_dev_ptrs = in GetrsBatchedImpl()
804 ScratchSpace<uint8> dev_b_dev_ptrs = in GetrsBatchedImpl()
847 ScratchSpace<uint8> dev_a_dev_ptrs = in GetriBatchedImpl()
850 ScratchSpace<uint8> dev_a_inv_dev_ptrs = cuda_solver->GetScratchSpace<uint8>( in GetriBatchedImpl()
890 ScratchSpace<uint8> dev_a_dev_ptrs = in MatInvBatchedImpl()
893 ScratchSpace<uint8> dev_a_inv_dev_ptrs = cuda_solver->GetScratchSpace<uint8>( in MatInvBatchedImpl()
986 ScratchSpace<uint8> dev_a_dev_ptrs = in TrsmBatchedImpl()
989 ScratchSpace<uint8> dev_b_dev_ptrs = in TrsmBatchedImpl()
/external/llvm-project/llvm/unittests/tools/llvm-exegesis/
DBenchmarkRunnerTest.cpp19 BenchmarkRunner::ScratchSpace Space; in TEST()
21 BenchmarkRunner::ScratchSpace::kAlignment, in TEST()
24 Space.ptr()[BenchmarkRunner::ScratchSpace::kSize - 1] = 43; in TEST()
27 EXPECT_EQ(Space.ptr()[BenchmarkRunner::ScratchSpace::kSize - 1], 0); in TEST()
/external/llvm-project/llvm/tools/llvm-exegesis/lib/
DBenchmarkRunner.h48 struct ScratchSpace { struct
51 ScratchSpace() in ScratchSpace() argument
87 const std::unique_ptr<ScratchSpace> Scratch;
DBenchmarkRunner.cpp34 : State(State), Mode(Mode), Scratch(std::make_unique<ScratchSpace>()) {} in BenchmarkRunner()
43 BenchmarkRunner::ScratchSpace *Scratch) in FunctionExecutorImpl()
131 BenchmarkRunner::ScratchSpace *const Scratch;
DParallelSnippetGenerator.cpp113 assert(I * MemStep < BenchmarkRunner::ScratchSpace::kSize && in instantiateMemoryOperands()
/external/llvm-project/llvm/unittests/tools/llvm-exegesis/Common/
DAssemblerUtils.h57 BenchmarkRunner::ScratchSpace Scratch; in Check()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_tensor_to_csr_sparse_matrix_op.cc139 ScratchSpace<int32> nnz_per_batch_host(c, batch_size, /*on_host*/ true); in ComputeAsync()
Ddense_to_csr_sparse_matrix_op.cc183 ScratchSpace<int32> nnz_per_batch_host(c, batch_size, /*on_host*/ true); in ComputeAsync()
/external/tensorflow/tensorflow/core/kernels/
Dwhere_op.cc296 ScratchSpace<Tindex> num_true_host(context, 1, /* on_host */ true); in ComputeAsyncType()
Dsegment_reduction_ops_impl.h261 ScratchSpace<Index> output_rows_host(context, 1, /* on_host */ true); in ComputeAsync()