OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:SyclDevice
(Results
1 - 25
of
85
) sorted by null
1
2
3
4
/external/tensorflow/tensorflow/core/kernels/
slice_op_cpu_impl.h
37
using
SyclDevice
= Eigen::
SyclDevice
;
40
template struct functor::Slice<
SyclDevice
, T, CPU_PROVIDED_IXDIM>;
snapshot_op.cc
35
typedef Eigen::
SyclDevice
SyclDevice
;
39
SnapshotOp<
SyclDevice
, TYPE>);
fill_functor.h
49
// Partial specialization of SetZeroFunctor<Device=Eigen::
SyclDevice
, T>.
51
struct SetZeroFunctor<Eigen::
SyclDevice
, T> {
52
void operator()(const Eigen::
SyclDevice
& d, typename TTypes<T>::Flat out);
76
// Partial specialization of SetOneFunctor<Device=Eigen::
SyclDevice
, T>.
78
struct SetOneFunctor<Eigen::
SyclDevice
, T> {
79
void operator()(const Eigen::
SyclDevice
& d, typename TTypes<T>::Flat out);
fill_functor.cc
61
void SetZeroFunctor<Eigen::
SyclDevice
, T>::operator()(
62
const Eigen::
SyclDevice
& d, typename TTypes<T>::Flat out) {
67
template struct SetZeroFunctor<Eigen::
SyclDevice
, T>;
106
void SetOneFunctor<Eigen::
SyclDevice
, T>::operator()(
107
const Eigen::
SyclDevice
& d, typename TTypes<T>::Flat out) {
112
template struct SetOneFunctor<Eigen::
SyclDevice
, T>;
139
struct FillFunctor<Eigen::
SyclDevice
, T> {
140
void operator()(const Eigen::
SyclDevice
& d, typename TTypes<T>::Flat out,
154
#define DEFINE_FILL_SYCL(T) template struct FillFunctor<Eigen::
SyclDevice
, T>;
split_lib_cpu.cc
47
void Split<Eigen::
SyclDevice
, T>::operator()(
48
const Eigen::
SyclDevice
& d, typename TTypes<T, 3>::Tensor output,
55
#define DEFINE_SYCL_KERNELS(T) template struct Split<Eigen::
SyclDevice
, T>;
cast_op_impl_bool.cc
38
typedef Eigen::
SyclDevice
SYCLDevice
;
41
CURRY_TYPES3_NO_HALF(CAST_CASE,
SYCLDevice
, bool);
cast_op_impl_double.cc
38
typedef Eigen::
SyclDevice
SYCLDevice
;
41
CURRY_TYPES3_NO_HALF(CAST_CASE,
SYCLDevice
, double);
cast_op_impl_int16.cc
38
typedef Eigen::
SyclDevice
SYCLDevice
;
41
CURRY_TYPES3_NO_HALF(CAST_CASE,
SYCLDevice
, int16);
cast_op_impl_int32.cc
38
typedef Eigen::
SyclDevice
SYCLDevice
;
41
CURRY_TYPES3_NO_HALF(CAST_CASE,
SYCLDevice
, int32);
cast_op_impl_int64.cc
38
typedef Eigen::
SyclDevice
SYCLDevice
;
41
CURRY_TYPES3_NO_HALF(CAST_CASE,
SYCLDevice
, int64);
cast_op_impl_int8.cc
38
typedef Eigen::
SyclDevice
SYCLDevice
;
41
CURRY_TYPES3_NO_HALF(CAST_CASE,
SYCLDevice
, int8);
cast_op_impl_uint16.cc
38
typedef Eigen::
SyclDevice
SYCLDevice
;
41
CURRY_TYPES3_NO_HALF(CAST_CASE,
SYCLDevice
, uint16);
cast_op_impl_uint8.cc
38
typedef Eigen::
SyclDevice
SYCLDevice
;
41
CURRY_TYPES3_NO_HALF(CAST_CASE,
SYCLDevice
, uint8);
split_lib.h
53
struct Split<Eigen::
SyclDevice
, T> {
54
void operator()(const Eigen::
SyclDevice
& d,
concat_lib.h
62
const Eigen::
SyclDevice
& d,
random_op.h
58
typedef Eigen::
SyclDevice
SYCLDevice
;
61
struct FillPhiloxRandom<
SYCLDevice
, Distribution> {
62
void operator()(OpKernelContext* ctx, const
SYCLDevice
& d,
inplace_ops.cc
29
typedef Eigen::
SyclDevice
SyclDevice
;
64
Status DoParallelConcat(const
SyclDevice
& d, const Tensor& value, int32 loc,
70
return DoParallelConcatUpdate<
SyclDevice
, type>(d, value, loc, output);
181
ParallelConcatStart<
SyclDevice
, type>);
196
ParallelConcatUpdate<
SyclDevice
>);
/external/eigen/unsupported/test/
cxx11_tensor_device_sycl.cpp
23
void test_device_sycl(const Eigen::
SyclDevice
&sycl_device) {
29
Eigen::
SyclDevice
sycl_device(s);
cxx11_tensor_broadcast_sycl.cpp
24
using Eigen::
SyclDevice
;
28
static void test_broadcast_sycl(const Eigen::
SyclDevice
&sycl_device){
72
Eigen::
SyclDevice
sycl_device(s);
cxx11_tensor_reduction_sycl.cpp
25
static void test_full_reductions_sycl(const Eigen::
SyclDevice
& sycl_device) {
55
static void test_first_dim_reductions_sycl(const Eigen::
SyclDevice
& sycl_device) {
93
static void test_last_dim_reductions_sycl(const Eigen::
SyclDevice
&sycl_device) {
133
Eigen::
SyclDevice
sycl_device(s);
cxx11_tensor_forced_eval_sycl.cpp
25
void test_forced_eval_sycl(const Eigen::
SyclDevice
&sycl_device) {
68
Eigen::
SyclDevice
sycl_device(s);
/external/eigen/bench/tensors/
tensor_benchmarks_sycl.cc
9
using Eigen::
SyclDevice
;
30
Eigen::
SyclDevice
device(q); \
31
BenchmarkSuite<Eigen::
SyclDevice
, float> suite(device, N); \
/external/tensorflow/tensorflow/core/framework/
device_base.h
33
struct
SyclDevice
;
160
void set_eigen_sycl_device(Eigen::
SyclDevice
* d) { eigen_sycl_device_ = d; }
186
virtual const Eigen::
SyclDevice
* eigen_sycl_device() const {
237
Eigen::
SyclDevice
* eigen_sycl_device_ = nullptr;
/external/tensorflow/tensorflow/core/common_runtime/sycl/
sycl_allocator.h
55
Eigen::
SyclDevice
* getSyclDevice() { return sycl_device_; }
67
Eigen::
SyclDevice
* sycl_device_ GUARDED_BY(mu_); // owned
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorDeviceSycl.h
19
struct
SyclDevice
{
24
/// per pointer. The lifespan of the buffer now depends on the lifespan of
SyclDevice
.
28
template<typename dev_Selector>
SyclDevice
(dev_Selector s)
45
~
SyclDevice
() { deallocate_all(); }
Completed in 364 milliseconds
1
2
3
4