Searched refs:OutT (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_shuffle_op.cc | 49 template <class IntT, class InT, class OutT, class Random>
51 OutT output_mat, Random& uniform) { in IndexedShuffle()
|
| D | segment_reduction_ops.cc | 150 OutT; in Compute() typedef
169 OutT out_slice(out_slice_ptr, out_slice_shape); in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.h | 117 template <typename InT, typename OutT, typename CompT>
122 DeviceMemory<OutT> *c, int ldc, blas::ComputationType computation_type,
|
| D | rocm_blas.cc | 1697 template <typename InT, typename OutT, typename CompT>
1702 DeviceMemory<OutT> *c, int ldc, blas::ComputationType computation_type, in DoBlasGemmWithAlgorithmImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 119 template <typename InT, typename OutT, typename CompT>
124 const HostOrDeviceScalar<CompT> &beta, DeviceMemory<OutT> *c, int ldc,
|
| D | cuda_blas.cc | 1886 template <typename InT, typename OutT, typename CompT>
1891 const HostOrDeviceScalar<CompT> &beta, DeviceMemory<OutT> *c, int ldc, in DoBlasGemmWithAlgorithmImpl()
1961 GpuMemoryMutable(c), CUDADataType<OutT>::type, ldc, in DoBlasGemmWithAlgorithmImpl()
|
| /external/deqp/external/openglcts/modules/gl/ |
| D | gl4cGPUShaderFP64Tests.cpp | 13184 template <typename ResT, typename ArgT, typename OutT>
13188 typedef ResT (*functionPointer)(ArgT, OutT&);
13201 OutT* p_out = (OutT*)((glw::GLubyte*)result_dst + getResultOffset(1)); in call()
13216 OutT& second_arg = p_out[component * component_step_1]; in call()
|