| /external/tensorflow/tensorflow/core/kernels/ |
| D | fill_functor.cc | 30 void SetZeroFunctor<Eigen::ThreadPoolDevice, T>::operator()( in operator ()()
31 const Eigen::ThreadPoolDevice& d, typename TTypes<T>::Flat out) { in operator ()()
35 void SetZeroFunctor<Eigen::ThreadPoolDevice, tstring>::operator()( in operator ()()
36 const Eigen::ThreadPoolDevice& d, typename TTypes<tstring>::Flat out) { in operator ()()
42 template struct SetZeroFunctor<Eigen::ThreadPoolDevice, T>;
68 void SetOneFunctor<Eigen::ThreadPoolDevice, T>::operator()( in operator ()()
69 const Eigen::ThreadPoolDevice& d, typename TTypes<T>::Flat out) { in operator ()()
75 template struct SetOneFunctor<Eigen::ThreadPoolDevice, T>;
95 struct FillFunctor<Eigen::ThreadPoolDevice, T> {
96 void operator()(const Eigen::ThreadPoolDevice& d, in operator ()()
[all …]
|
| D | fill_functor.h | 43 struct SetZeroFunctor<Eigen::ThreadPoolDevice, T> {
44 void operator()(const Eigen::ThreadPoolDevice& d,
50 struct SetZeroFunctor<Eigen::ThreadPoolDevice, tstring> {
51 void operator()(const Eigen::ThreadPoolDevice& d,
63 struct SetOneFunctor<Eigen::ThreadPoolDevice, T> {
64 void operator()(const Eigen::ThreadPoolDevice& d,
70 struct SetOneFunctor<Eigen::ThreadPoolDevice, tstring> {
71 void operator()(const Eigen::ThreadPoolDevice& d,
|
| D | split_lib_cpu.cc | 28 void Split<Eigen::ThreadPoolDevice, T, NDims>::operator()( in operator ()()
29 const Eigen::ThreadPoolDevice& d, typename TTypes<T, NDims>::Tensor output, in operator ()()
41 template struct Split<Eigen::ThreadPoolDevice, T, 2>; \
42 template struct Split<Eigen::ThreadPoolDevice, T, 3>;
|
| D | tile_functor_cpu.h | 47 void TileSimple(const Eigen::ThreadPoolDevice& d, Tensor* out, in TileSimple()
49 return TileSimpleImpl<Eigen::ThreadPoolDevice, T>(d, out, in); in TileSimple()
|
| D | split_lib.h | 43 struct Split<Eigen::ThreadPoolDevice, T, NDims> {
44 void operator()(const Eigen::ThreadPoolDevice& d,
|
| D | eigen_benchmark_cpu_test.cc | 24 Eigen::ThreadPoolDevice device(&tp, threads)
39 SpatialConvolutionBenchmarksSuite<float, Eigen::ThreadPoolDevice>; in SpatialConvolution()
67 SpatialConvolutionBenchmarksSuite<float, Eigen::ThreadPoolDevice>; in SpatialConvolutionBackwardInput()
94 SpatialConvolutionBenchmarksSuite<float, Eigen::ThreadPoolDevice>; in SpatialConvolutionBackwardKernel()
257 CuboidConvolutionBenchmarksSuite<float, Eigen::ThreadPoolDevice>; in CuboidConvolution()
286 CuboidConvolutionBenchmarksSuite<float, Eigen::ThreadPoolDevice>; in CuboidConvolutionBackwardInput()
314 CuboidConvolutionBenchmarksSuite<float, Eigen::ThreadPoolDevice>; in CuboidConvolutionBackwardKernel()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | threadpool_device.cc | 45 ThreadPoolDevice::ThreadPoolDevice(const SessionOptions& options, in ThreadPoolDevice() function in tensorflow::ThreadPoolDevice
71 ThreadPoolDevice::~ThreadPoolDevice() {} in ~ThreadPoolDevice()
73 Allocator* ThreadPoolDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator()
77 Allocator* ThreadPoolDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator()
88 Status ThreadPoolDevice::MakeTensorFromProto( in MakeTensorFromProto()
102 void ThreadPoolDevice::CopyTensorInSameDevice( in CopyTensorInSameDevice()
|
| D | threadpool_device.h | 25 class ThreadPoolDevice : public LocalDevice {
27 ThreadPoolDevice(const SessionOptions& options, const string& name,
30 ~ThreadPoolDevice() override;
|
| D | threadpool_device_factory.cc | 48 std::unique_ptr<ThreadPoolDevice> tpd; in CreateDevices()
59 tpd = absl::make_unique<ThreadPoolDevice>( in CreateDevices()
63 tpd = absl::make_unique<ThreadPoolDevice>( in CreateDevices()
|
| D | mkl_threadpool_device_test.cc | 32 ThreadPoolDevice* tp = new ThreadPoolDevice( in TEST()
|
| D | single_threaded_cpu_device.cc | 50 eigen_device_.reset(new Eigen::ThreadPoolDevice( in SingleThreadedCpuDevice()
91 std::unique_ptr<Eigen::ThreadPoolDevice> eigen_device_;
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | eigen_support.cc | 101 const Eigen::ThreadPoolDevice* GetThreadPoolDevice() { in GetThreadPoolDevice()
105 device_.reset(new Eigen::ThreadPoolDevice(thread_pool_wrapper_.get(), in GetThreadPoolDevice()
125 std::unique_ptr<Eigen::ThreadPoolDevice> device_;
184 const Eigen::ThreadPoolDevice* GetThreadPoolDevice(TfLiteContext* context) { in GetThreadPoolDevice()
|
| D | eigen_support.h | 21 struct ThreadPoolDevice;
40 const EigenForTFLite::ThreadPoolDevice* GetThreadPoolDevice(
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.h | 34 struct ThreadPoolDevice;
119 const Eigen::ThreadPoolDevice* intra_op_thread_pool);
120 const Eigen::ThreadPoolDevice* intra_op_thread_pool() const;
163 const Eigen::ThreadPoolDevice* intra_op_thread_pool_ = nullptr;
|
| D | executable_run_options.cc | 75 const Eigen::ThreadPoolDevice* intra_op_thread_pool) { in set_intra_op_thread_pool()
80 const Eigen::ThreadPoolDevice* ExecutableRunOptions::intra_op_thread_pool() in intra_op_thread_pool()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_thread_pool.cpp | 30 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(3, 11)); in test_multithread_elementwise()
53 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(3, 11)); in test_multithread_compound_assignment()
86 Eigen::ThreadPoolDevice thread_pool_device(&tp, 4); in test_multithread_contraction()
127 Eigen::ThreadPoolDevice thread_pool_device(&tp, 12); in test_contraction_corner_cases()
217 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(2, 11)); in test_multithread_contraction_agrees_with_singlethread()
256 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(2, 11)); in test_full_contraction()
276 Eigen::ThreadPoolDevice thread_pool_device(&thread_pool, num_threads); in test_multithreaded_reductions()
300 Eigen::ThreadPoolDevice thread_pool_device(&tp, num_threads); in test_memcpy()
317 Eigen::ThreadPoolDevice device(&tp, 2); in test_multithread_random()
330 Eigen::ThreadPoolDevice device(&threads, num_threads); in test_multithread_shuffle()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.cc | 76 void DeviceBase::set_eigen_cpu_device(Eigen::ThreadPoolDevice* d) { in set_eigen_cpu_device()
83 eigen_cpu_devices_.push_back(new Eigen::ThreadPoolDevice( in set_eigen_cpu_device()
88 const Eigen::ThreadPoolDevice* DeviceBase::eigen_cpu_device() { in eigen_cpu_device()
|
| D | device_base.h | 34 struct ThreadPoolDevice;
177 void set_eigen_cpu_device(Eigen::ThreadPoolDevice* d);
207 virtual const Eigen::ThreadPoolDevice* eigen_cpu_device();
281 std::vector<Eigen::ThreadPoolDevice*> eigen_cpu_devices_;
|
| /external/eigen/bench/tensors/ |
| D | contraction_benchmarks_cpu.cc | 9 Eigen::ThreadPoolDevice device(&pool, threads);
18 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, D1, D2, D3); \
|
| D | tensor_benchmarks_cpu.cc | 9 Eigen::ThreadPoolDevice device(&pool, threads);
16 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, N); \
92 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, D1, D2, D3); \
141 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, N); \
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 87 void operator()(const Eigen::ThreadPoolDevice& device, const T* input_data, in operator()
106 MatMulConvFunctor<Eigen::ThreadPoolDevice, T>()(device, output, input, in operator()
119 MatMulConvFunctor<Eigen::ThreadPoolDevice, T>()(device, output, input, in operator()
135 inline void Conv(const Eigen::ThreadPoolDevice& device, in Conv()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | test.cc | 63 Eigen::ThreadPoolDevice device(&pool, pool.NumThreads()); in TEST()
77 Eigen::ThreadPoolDevice device(&pool, pool.NumThreads()); in BM_NAME()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_factory.cc | 81 class GPUCompatibleCPUDevice : public ThreadPoolDevice {
86 : ThreadPoolDevice(options, name, memory_limit, locality, allocator), in GPUCompatibleCPUDevice()
101 return ThreadPoolDevice::GetAllocator(attr); in GetAllocator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 75 device(new Eigen::ThreadPoolDevice(pool->AsEigenThreadPool(), in IntraOpThreadPool()
79 std::unique_ptr<Eigen::ThreadPoolDevice> device;
151 const Eigen::ThreadPoolDevice* Backend::eigen_intra_op_thread_pool_device() in eigen_intra_op_thread_pool_device()
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 247 Eigen::ThreadPoolDevice device(&tp, tp.NumThreads()); in TEST()
318 Eigen::ThreadPoolDevice device(&tp, tp.NumThreads()); in TEST()
344 Eigen::ThreadPoolDevice device(&tp, tp.NumThreads()); in TEST()
450 Eigen::ThreadPoolDevice device(&tp, tp.NumThreads()); in TEST()
478 Eigen::ThreadPoolDevice device(&tp, tp.NumThreads()); in TEST()
502 Eigen::ThreadPoolDevice device(&tp, tp.NumThreads()); in TEST()
516 Eigen::ThreadPoolDevice device(&tp, tp.NumThreads()); in TEST()
538 Eigen::ThreadPoolDevice device(&tp, tp.NumThreads()); in TEST()
565 Eigen::ThreadPoolDevice device(&tp, tp.NumThreads()); in TEST()
661 Eigen::ThreadPoolDevice device(&tp, tp.NumThreads()); in TEST()
|