Home
last modified time | relevance | path

Searched refs:DefaultDevice (Results 1 – 25 of 35) sorted by relevance

12

/external/eigen/unsupported/test/
Dcxx11_tensor_chipping.cpp348 typedef TensorEvaluator<decltype(tensor.chip<4>(3)), DefaultDevice> Evaluator4; in test_chip_raw_data_col_major()
349 auto chip = Evaluator4(tensor.chip<4>(3), DefaultDevice()); in test_chip_raw_data_col_major()
361 typedef TensorEvaluator<decltype(tensor.chip<0>(0)), DefaultDevice> Evaluator0; in test_chip_raw_data_col_major()
362 auto chip0 = Evaluator0(tensor.chip<0>(0), DefaultDevice()); in test_chip_raw_data_col_major()
365 typedef TensorEvaluator<decltype(tensor.chip<1>(0)), DefaultDevice> Evaluator1; in test_chip_raw_data_col_major()
366 auto chip1 = Evaluator1(tensor.chip<1>(0), DefaultDevice()); in test_chip_raw_data_col_major()
369 typedef TensorEvaluator<decltype(tensor.chip<2>(0)), DefaultDevice> Evaluator2; in test_chip_raw_data_col_major()
370 auto chip2 = Evaluator2(tensor.chip<2>(0), DefaultDevice()); in test_chip_raw_data_col_major()
373 typedef TensorEvaluator<decltype(tensor.chip<3>(0)), DefaultDevice> Evaluator3; in test_chip_raw_data_col_major()
374 auto chip3 = Evaluator3(tensor.chip<3>(0), DefaultDevice()); in test_chip_raw_data_col_major()
[all …]
Dcxx11_tensor_morphing.cpp81 tensor5d.reshape(dim).device(Eigen::DefaultDevice()) = tensor; in test_reshape_as_lvalue()
185 result.slice(second_slice, sizes12).device(Eigen::DefaultDevice()) = tensor2; in test_slice_as_lvalue()
234 typedef TensorEvaluator<decltype(tensor.slice(offsets, extents)), DefaultDevice> SliceEvaluator; in test_slice_raw_data()
235 auto slice1 = SliceEvaluator(tensor.slice(offsets, extents), DefaultDevice()); in test_slice_raw_data()
241 auto slice2 = SliceEvaluator(tensor.slice(offsets, extents), DefaultDevice()); in test_slice_raw_data()
247 auto slice2 = SliceEvaluator(tensor.slice(offsets, extents), DefaultDevice()); in test_slice_raw_data()
254 auto slice3 = SliceEvaluator(tensor.slice(offsets, extents), DefaultDevice()); in test_slice_raw_data()
261 auto slice4 = SliceEvaluator(tensor.slice(offsets, extents), DefaultDevice()); in test_slice_raw_data()
271 auto slice4 = SliceEvaluator(tensor.slice(offsets, extents), DefaultDevice()); in test_slice_raw_data()
283 auto slice5 = SliceEvaluator(tensor.slice(offsets, extents), DefaultDevice()); in test_slice_raw_data()
[all …]
Dcxx11_tensor_contraction.cpp14 using Eigen::DefaultDevice;
33 typedef TensorEvaluator<decltype(mat1.contract(mat2, dims3)), DefaultDevice> Evaluator; in test_evals()
34 Evaluator eval(mat1.contract(mat2, dims3), DefaultDevice()); in test_evals()
53 typedef TensorEvaluator<decltype(mat1.contract(mat2, dims4)), DefaultDevice> Evaluator2; in test_evals()
54 Evaluator2 eval2(mat1.contract(mat2, dims4), DefaultDevice()); in test_evals()
68 typedef TensorEvaluator<decltype(mat1.contract(mat3, dims6)), DefaultDevice> Evaluator3; in test_evals()
69 Evaluator3 eval3(mat1.contract(mat3, dims6), DefaultDevice()); in test_evals()
112 typedef TensorEvaluator<decltype(mat1.contract(mat2, dims)), DefaultDevice> Evaluator; in test_multidims()
113 Evaluator eval(mat1.contract(mat2, dims), DefaultDevice()); in test_multidims()
146 typedef TensorEvaluator<decltype(mat4.contract(mat5, dims2)), DefaultDevice> Evaluator2; in test_multidims()
[all …]
Dcxx11_tensor_convolution.cpp15 using Eigen::DefaultDevice;
30 typedef TensorEvaluator<decltype(input.convolve(kernel, dims3)), DefaultDevice> Evaluator; in test_evals()
31 Evaluator eval(input.convolve(kernel, dims3), DefaultDevice()); in test_evals()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensor.h388 resize(TensorEvaluator<const Assign, DefaultDevice>(assign, DefaultDevice()).dimensions()); in Tensor()
389 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice()); in Tensor()
397 resize(TensorEvaluator<const Assign, DefaultDevice>(assign, DefaultDevice()).dimensions()); in Tensor()
398 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice()); in Tensor()
406 resize(TensorEvaluator<const Assign, DefaultDevice>(assign, DefaultDevice()).dimensions());
407 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
416 resize(TensorEvaluator<const Assign, DefaultDevice>(assign, DefaultDevice()).dimensions());
417 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
DTensorInitializer.h31 static void run(TensorEvaluator<Derived, DefaultDevice>& tensor, in run()
46 static void run(TensorEvaluator<Derived, DefaultDevice>& tensor,
62 static void run(TensorEvaluator<Derived, DefaultDevice>& tensor,
71 void initialize_tensor(TensorEvaluator<Derived, DefaultDevice>& tensor,
DTensorFixedSize.h325 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice()); in TensorFixedSize()
333 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice()); in TensorFixedSize()
343 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
354 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
DTensorExecutor.h49 class TensorExecutor<Expression, DefaultDevice, true>
54 static inline void run(const Expression& expr, const DefaultDevice& device = DefaultDevice())
56 TensorEvaluator<Expression, DefaultDevice> evaluator(expr, device);
61 …const int PacketSize = unpacket_traits<typename TensorEvaluator<Expression, DefaultDevice>::Packet…
DTensorLayoutSwap.h90 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
100 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
DTensorReductionSycl.h159 …ensorEvaluator<decltype(device_self_expr), Eigen::DefaultDevice>(device_self_expr, Eigen::DefaultD…
222 typedef Eigen::TensorEvaluator<decltype(device_self_expr), Eigen::DefaultDevice> DeiceSelf;
223 …ensorEvaluator<decltype(device_self_expr), Eigen::DefaultDevice>(device_self_expr, Eigen::DefaultD…
DTensorSyclRun.h56 …ensorEvaluator<decltype(device_expr.expr), Eigen::DefaultDevice>(device_expr.expr, Eigen::DefaultD… in run()
DTensorMorphing.h77 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
87 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
263 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
272 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
668 internal::TensorExecutor<const Assign, DefaultDevice>::run(
669 assign, DefaultDevice());
679 internal::TensorExecutor<const Assign, DefaultDevice>::run(
680 assign, DefaultDevice());
DTensorIO.h60 typedef TensorEvaluator<const TensorForcedEvalOp<const T>, DefaultDevice> Evaluator;
65 Evaluator tensor(eval, DefaultDevice());
DTensorReverse.h80 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
90 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
DTensorShuffling.h78 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
88 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
DTensorMap.h300 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
310 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
DTensorStriding.h78 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
88 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
DTensorRef.h148 …(new internal::TensorLazyEvaluator<Dimensions, Expression, DefaultDevice>(expr, DefaultDevice())) { in TensorRef()
155 … = new internal::TensorLazyEvaluator<Dimensions, Expression, DefaultDevice>(expr, DefaultDevice());
DTensorConcatenation.h87 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
97 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
DTensorChipping.h106 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
116 internal::TensorExecutor<const Assign, DefaultDevice>::run(assign, DefaultDevice());
/external/tensorflow/tensorflow/core/kernels/
Dunary_ops_composition.cc176 auto relu = functor::Relu<Eigen::DefaultDevice, T>(); \
177 relu(Eigen::DefaultDevice(), in, *out); \
185 auto relu6 = functor::Relu6<Eigen::DefaultDevice, T>(); \
186 relu6(Eigen::DefaultDevice(), in, *out); \
194 auto elu = functor::Elu<Eigen::DefaultDevice, T>(); \
195 elu(Eigen::DefaultDevice(), in, *out); \
203 auto selu = functor::Selu<Eigen::DefaultDevice, T>(); \
204 selu(Eigen::DefaultDevice(), in, *out); \
/external/deqp/external/vulkancts/modules/vulkan/
DvktTestCase.cpp459 class DefaultDevice class
462DefaultDevice (const PlatformInterface& vkPlatform, const tcu::CommandLine& cmdLine);
463 ~DefaultDevice (void);
527 DefaultDevice::DefaultDevice (const PlatformInterface& vkPlatform, const tcu::CommandLine& cmdLine) in DefaultDevice() function in vkt::DefaultDevice
550 DefaultDevice::~DefaultDevice (void) in ~DefaultDevice()
554 VkQueue DefaultDevice::getUniversalQueue (void) const in getUniversalQueue()
559 VkQueue DefaultDevice::getSparseQueue (void) const in getSparseQueue()
571 vk::Allocator* createAllocator (DefaultDevice* device) in createAllocator()
601 , m_device (new DefaultDevice(m_platformInterface, testCtx.getCommandLine())) in Context()
DvktTestCase.hpp52 class DefaultDevice;
119 const de::UniquePtr<DefaultDevice> m_device;
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc35 Eigen::DefaultDevice(), out, lhs, rhs, input_batch, input_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
52 Eigen::DefaultDevice(), out, lhs, rhs, input_batch, input_rows, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_single_threaded_fft.cc29 tensorflow::xla::EigenFftImpl(Eigen::DefaultDevice(), out, operand, fft_type, in __xla_cpu_runtime_EigenSingleThreadedFft()

12