Searched refs:InT (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.h | 117 template <typename InT, typename OutT, typename CompT>
120 uint64 n, uint64 k, const CompT &alpha, const DeviceMemory<InT> &a,
121 int lda, const DeviceMemory<InT> &b, int ldb, const CompT &beta,
|
| D | rocm_blas.cc | 1697 template <typename InT, typename OutT, typename CompT>
1700 uint64 n, uint64 k, const CompT &alpha, const DeviceMemory<InT> &a, int lda, in DoBlasGemmWithAlgorithmImpl()
1701 const DeviceMemory<InT> &b, int ldb, const CompT &beta, in DoBlasGemmWithAlgorithmImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 119 template <typename InT, typename OutT, typename CompT>
123 const DeviceMemory<InT> &a, int lda, const DeviceMemory<InT> &b, int ldb,
|
| D | cuda_blas.cc | 1886 template <typename InT, typename OutT, typename CompT>
1890 const DeviceMemory<InT> &a, int lda, const DeviceMemory<InT> &b, int ldb, in DoBlasGemmWithAlgorithmImpl()
1904 if (UsesTensorOps(algorithm) && !TensorOpsAvailable<InT>(cc_major)) { in DoBlasGemmWithAlgorithmImpl()
1905 if (std::is_same<InT, Eigen::half>::value) { in DoBlasGemmWithAlgorithmImpl()
1948 cudaDataType_t cuda_in_type = CUDADataType<InT>::type; in DoBlasGemmWithAlgorithmImpl()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_shuffle_op.cc | 49 template <class IntT, class InT, class OutT, class Random>
50 static void IndexedShuffle(const int64 size, const InT& input_mat, in IndexedShuffle()
|
| D | segment_reduction_ops.cc | 177 InT; in Compute() typedef
178 InT in_slice(in_slice_ptr, out_slice_shape); in Compute()
185 InT; in Compute() typedef
186 InT in_slice(in_slice_ptr, in_slice_shape); in Compute()
|