Home
last modified time | relevance | path

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

/external/eigen/unsupported/test/
Dcxx11_tensor_cuda.cu81 float* d_out; in test_cuda_elementwise_small() local
84 cudaMalloc((void**)(&d_out), out_bytes); in test_cuda_elementwise_small()
97 d_out, Eigen::array<Eigen::DenseIndex, 1>(2)); in test_cuda_elementwise_small()
101 assert(cudaMemcpyAsync(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, in test_cuda_elementwise_small()
113 cudaFree(d_out); in test_cuda_elementwise_small()
134 float* d_out; in test_cuda_elementwise() local
138 cudaMalloc((void**)(&d_out), out_bytes); in test_cuda_elementwise()
150 …Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_out(d_out, Eigen::array<Eigen::DenseIndex, 3>(72,53… in test_cuda_elementwise()
154 …assert(cudaMemcpyAsync(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, gpu_device.stream()) … in test_cuda_elementwise()
168 cudaFree(d_out); in test_cuda_elementwise()
[all …]
Dcxx11_tensor_random_cuda.cu30 float* d_out; in test_cuda_random_uniform() local
31 cudaMalloc((void**)(&d_out), out_bytes); in test_cuda_random_uniform()
36 Eigen::TensorMap<Eigen::Tensor<float, 2> > gpu_out(d_out, 72,97); in test_cuda_random_uniform()
40 …assert(cudaMemcpyAsync(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, gpu_device.stream()) … in test_cuda_random_uniform()
55 float* d_out; in test_cuda_random_normal() local
56 cudaMalloc((void**)(&d_out), out_bytes); in test_cuda_random_normal()
61 Eigen::TensorMap<Eigen::Tensor<float, 2> > gpu_out(d_out, 72,97); in test_cuda_random_normal()
66 …assert(cudaMemcpyAsync(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, gpu_device.stream()) … in test_cuda_random_normal()
Dcxx11_tensor_argmax_cuda.cu102 DenseIndex* d_out; in test_cuda_argmax_dim() local
104 cudaMalloc((void**)(&d_out), out_bytes); in test_cuda_argmax_dim()
112 Eigen::TensorMap<Eigen::Tensor<DenseIndex, 3, DataLayout>, Aligned > gpu_out(d_out, out_shape); in test_cuda_argmax_dim()
116 …assert(cudaMemcpyAsync(tensor_arg.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, gpu_device.str… in test_cuda_argmax_dim()
144 …assert(cudaMemcpyAsync(tensor_arg.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, gpu_device.str… in test_cuda_argmax_dim()
153 cudaFree(d_out); in test_cuda_argmax_dim()
191 DenseIndex* d_out; in test_cuda_argmin_dim() local
193 cudaMalloc((void**)(&d_out), out_bytes); in test_cuda_argmin_dim()
201 Eigen::TensorMap<Eigen::Tensor<DenseIndex, 3, DataLayout>, Aligned > gpu_out(d_out, out_shape); in test_cuda_argmin_dim()
205 …assert(cudaMemcpyAsync(tensor_arg.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, gpu_device.str… in test_cuda_argmin_dim()
[all …]
Dcxx11_tensor_complex_cwise_ops_cuda.cu29 std::complex<T>* d_out; in test_cuda_complex_cwise_ops() local
32 cudaMalloc((void**)(&d_out), complex_bytes); in test_cuda_complex_cwise_ops()
42 d_out, kNumItems); in test_cuda_complex_cwise_ops()
78 assert(cudaMemcpyAsync(actual.data(), d_out, complex_bytes, cudaMemcpyDeviceToHost, in test_cuda_complex_cwise_ops()
89 cudaFree(d_out); in test_cuda_complex_cwise_ops()
Dcxx11_tensor_device.cu285 float* d_out; in test_gpu() local
288 cudaMalloc((void**)(&d_out), out_bytes); in test_gpu()
295 Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_out(d_out, 40,50,70); in test_gpu()
299 assert(cudaMemcpy(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost) == cudaSuccess); in test_gpu()
309 assert(cudaMemcpy(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost) == cudaSuccess); in test_gpu()
319 assert(cudaMemcpy(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost) == cudaSuccess); in test_gpu()
329 assert(cudaMemcpy(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost) == cudaSuccess); in test_gpu()
344 …assert(cudaMemcpyAsync(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, context.device().stre… in test_gpu()
355 …assert(cudaMemcpyAsync(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, context.device().stre… in test_gpu()
369 …assert(cudaMemcpyAsync(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost, context.device().stre… in test_gpu()
/external/eigen/test/
Dcuda_common.h37 typename Output::Scalar* d_out; in run_on_cuda() local
42 cudaMalloc((void**)(&d_out), out_bytes); in run_on_cuda()
45 cudaMemcpy(d_out, out.data(), out_bytes, cudaMemcpyHostToDevice); in run_on_cuda()
53 run_on_cuda_meta_kernel<<<Grids,Blocks>>>(ker, n, d_in, d_out); in run_on_cuda()
58 cudaMemcpy(out.data(), d_out, out_bytes, cudaMemcpyDeviceToHost); in run_on_cuda()
61 cudaFree(d_out); in run_on_cuda()
/external/tensorflow/tensorflow/core/kernels/
Dwhere_op_gpu.cu.h73 const T* d_in, TIndex* d_out, int num_items, in operator()
80 is_nonzero_iter, d_out, num_items, stream, in operator()
88 const bool* d_in, TIndex* d_out, int num_items,
92 d_out, num_items, stream, debug_synchronous);
104 const T* d_flags, OutputIterator d_out,
114 is_nonzero_iter /*d_flags*/, d_out, d_num_selected_out, num_items,
123 const T* d_flags, OutputIterator d_out,
130 d_out, d_num_selected_out, num_items, stream, debug_synchronous);
/external/tensorflow/tensorflow/contrib/eager/python/examples/spinn/
Dspinn_test.py71 def _test_spinn_config(d_embed, d_out, logdir=None, inference_sentences=None): argument
104 d_out=d_out,
273 d_out = 4
275 config = _test_spinn_config(d_embed, d_out)
291 self.assertEqual((batch_size, d_out), logits.shape)
296 self.assertEqual((batch_size, d_out), logits.shape)
308 self.assertEqual((batch_size, d_out), logits.shape)
448 d_out = 4
452 config = _test_spinn_config(d_embed, d_out)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dmasked_autoregressive.py326 d_out = n_out // num_blocks
327 row = d_out if mask_type == MASK_EXCLUSIVE else 0
333 row += d_out
/external/tensorflow/tensorflow/python/framework/
Dsubscribe_test.py72 d_out = self.evaluate([d])
76 self.assertEqual(d_out, [42])
152 d_out = self.evaluate([d])
155 self.assertEqual(d_out, [11])
/external/freetype/src/base/
Dftcalc.c1049 FT_Pos d_in, d_out, d_hypot; in ft_corner_is_flat() local
1072 d_out = FT_HYPOT( out_x, out_y ); in ft_corner_is_flat()
1079 return ( d_in + d_out - d_hypot ) < ( d_hypot >> 4 ); in ft_corner_is_flat()
/external/epid-sdk/epid/common/math/src/
Dpairing.c87 static EpidStatus FrobeniusOp(PairingState* ps, FfElement* d_out,
871 static EpidStatus FrobeniusOp(PairingState* ps, FfElement* d_out, in FrobeniusOp() argument
881 if (!ps || !d_out || !a) { in FrobeniusOp()
887 if (e < 1 || e > 3 || !d_out->ipp_ff_elem || !a->ipp_ff_elem || in FrobeniusOp()
969 d_out->ipp_ff_elem, ps->ff->ipp_ff); in FrobeniusOp()