| /external/gemmlowp/meta/ |
| D | transform_kernels.h | 70 template <typename InType, typename OutType, int kernel_size, int leftovers>
71 class Transform1DKernel<InType, OutType, Quantize, kernel_size, leftovers> {
74 OutType* output) { in Transform()
78 << ", " << std::string(typeid(OutType).name()) << ") -- " in Transform()
88 template <typename InType, typename OutType, int kernel_size, int leftovers>
89 class Transform1DKernel<InType, OutType, Dequantize, kernel_size, leftovers> {
92 OutType* output) { in Transform()
96 << ", " << std::string(typeid(OutType).name()) << ") -- " in Transform()
106 template <typename InType, typename OutType, int kernel_size, int leftovers>
107 class Transform1DKernel<InType, OutType, Requantize, kernel_size, leftovers> {
[all …]
|
| D | quantized_mul_kernels.h | 35 template <typename InType, typename OutType, int m, int n, int k>
36 class MulKernel<InType, OutType, QuantizedStaticPreprocessed, RowMajor, m, n,
42 const FusedKernel& params, OutType* result) { in Multiply()
46 << typeid(OutType).name() << ")::Multiply() -- " << m << "x" << n in Multiply()
62 << typeid(OutType).name() << ") -- " << m << "x" << n << "x" << k in Debug()
83 template <typename InType, typename OutType, int m, int n, int k>
84 class MulKernel<InType, OutType, QuantizedStaticPreprocessedAsInt32, RowMajor,
91 const FusedKernel& params, OutType* result) { in Multiply()
95 << typeid(OutType).name() << ")::Multiply() -- " << m << "x" << n in Multiply()
111 << typeid(OutType).name() << ") -- " << m << "x" << n << "x" << k in Debug()
[all …]
|
| D | base.h | 50 typedef OutType_ OutType; typedef
66 OutType* result;
100 template <typename InType, typename OutType, typename Kernel,
106 OutType* result);
112 typedef OutType_ OutType; typedef
116 OutType* output;
122 template <typename InType, typename OutType, typename Kernel, int kernel_size,
127 OutType* output);
130 template <typename InType, typename OutType, typename Transform>
138 static OutType* OffsetOutput(const Transform& params, OutType* output,
|
| D | single_thread_gemm.h | 49 typedef typename P::OutType OutType; in ExecuteDispatch3D() typedef
65 typedef Stream<typename P::OutType, m, n, 0, typename P::OutputStream> in ExecuteDispatch3D()
67 typedef Stream<typename P::OutType, m_leftovers, n, 0, in ExecuteDispatch3D()
71 typedef MulKernel<typename P::InType, typename P::OutType, in ExecuteDispatch3D()
74 typedef MulKernel<typename P::InType, typename P::OutType, in ExecuteDispatch3D()
78 typedef MulKernel<typename P::InType, typename P::OutType, in ExecuteDispatch3D()
82 typedef MulKernel<typename P::InType, typename P::OutType, in ExecuteDispatch3D()
171 reinterpret_cast<OutType*>(result_chunk)); in ExecuteDispatch3D()
180 reinterpret_cast<OutType*>(result_chunk)); in ExecuteDispatch3D()
203 reinterpret_cast<OutType*>(result_chunk)); in ExecuteDispatch3D()
[all …]
|
| D | single_thread_transform.h | 33 Transform1DKernel<typename P::InType, typename P::OutType, in ExecuteDispatch1D()
|
| D | multi_thread_transform.h | 32 typedef Transform1DUtil<typename Params::InType, typename Params::OutType, in PrepareTransform1DTasks()
|
| D | multi_thread_gemm.h | 48 StreamUtil<typename Params::OutType, typename Params::OutputStream>:: in PrepareGemmTask()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | searchsorted_op_gpu.cu.cc | 33 template <typename T, typename OutType>
37 OutType* __restrict__ outputs) { in UpperBoundKernel()
41 outputs[work_unit_id] = gpu_helper::upper_bound<T, OutType>( in UpperBoundKernel()
46 template <typename T, typename OutType>
50 OutType* __restrict__ outputs) { in LowerBoundKernel()
54 outputs[work_unit_id] = gpu_helper::lower_bound<T, OutType>( in LowerBoundKernel()
61 template <typename T, typename OutType>
62 struct UpperBoundFunctor<GPUDevice, T, OutType> {
67 typename TTypes<OutType, 1>::Tensor* output) { in Compute()
76 UpperBoundKernel<T, OutType>, config.block_count, in Compute()
[all …]
|
| D | searchsorted_op.cc | 33 template <typename T, typename OutType>
34 struct UpperBoundFunctor<CPUDevice, T, OutType> {
39 typename TTypes<OutType, 1>::Tensor* output) { in Compute()
43 OutType* output_ptr = output->data() + b * num_values; in Compute()
56 template <typename T, typename OutType>
57 struct LowerBoundFunctor<CPUDevice, T, OutType> {
62 typename TTypes<OutType, 1>::Tensor* output) { in Compute()
66 OutType* output_ptr = output->data() + b * num_values; in Compute()
80 template <typename Device, typename T, typename OutType>
111 auto output = output_t->template flat<OutType>(); in Compute()
[all …]
|
| D | searchsorted_op.h | 28 template <typename Device, typename T, typename OutType>
36 typename TTypes<OutType, 1>::Tensor* output);
39 template <typename Device, typename T, typename OutType>
47 typename TTypes<OutType, 1>::Tensor* output);
|
| D | shape_ops.h | 41 template <typename OutType>
52 auto vec = out->vec<OutType>(); in Compute()
61 vec(i) = static_cast<OutType>(dim_size); in Compute()
68 template <typename OutType>
80 auto vec = out->vec<OutType>(); in Compute()
90 vec(j) = static_cast<OutType>(dim_size); in Compute()
114 template <typename OutType>
131 out->scalar<OutType>()() = static_cast<OutType>(size); in Compute()
|
| D | sparse_cross_op.cc | 260 template <typename OutType>
270 const OutType& cross) const { in Update()
277 auto value_vec = values_out_->vec<OutType>(); in Update()
|
| /external/libaom/libaom/test/ |
| D | transform_test_base.h | 40 template <typename OutType>
46 virtual void RunFwdTxfm(const int16_t *in, OutType *out, int stride) = 0;
48 virtual void RunInvTxfm(const OutType *out, uint8_t *dst, int stride) = 0;
58 OutType *test_temp_block = reinterpret_cast<OutType *>( in RunAccuracyCheck()
129 OutType *output_ref_block = reinterpret_cast<OutType *>( in RunCoeffCheck()
131 OutType *output_block = reinterpret_cast<OutType *>( in RunCoeffCheck()
178 OutType *trans_block = reinterpret_cast<OutType *>( in RunInvCoeffCheck()
224 OutType *output_ref_block = reinterpret_cast<OutType *>( in RunMemCheck()
226 OutType *output_block = reinterpret_cast<OutType *>( in RunMemCheck()
266 OutType *coeff = reinterpret_cast<OutType *>( in RunInvAccuracyCheck()
[all …]
|
| /external/angle/src/libANGLE/ |
| D | queryconversions.h | 84 template <typename OutType>
85 OutType ConvertGLenum(GLenum param)
87 return static_cast<OutType>(param);
90 template <typename InType, typename OutType>
91 void ConvertGLenumValue(InType param, OutType *out)
93 *out = ConvertGLenum<OutType>(static_cast<GLenum>(param));
96 template <typename PackedEnumType, typename OutType>
97 void ConvertPackedEnum(PackedEnumType param, OutType *out)
99 *out = static_cast<OutType>(ToGLenum(param));
|
| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_kernel_helper.h | 200 template <typename T, typename OutType = int32>
201 __device__ OutType upper_bound(const T* first, OutType count, T val) { in upper_bound()
204 OutType step = 0; in upper_bound()
220 template <typename T, typename OutType = int32>
221 __device__ OutType lower_bound(const T* first, OutType count, T val) { in lower_bound()
224 OutType step = 0; in lower_bound()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | shape.cc | 32 template <typename OutType>
33 void ExtractShape(const TfLiteTensor* input, OutType* output_data) { in ExtractShape()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | bfloat16.cc | 714 template <typename InType, typename OutType, typename Functor>
717 return {TypeDescriptor<InType>::Dtype(), TypeDescriptor<OutType>::Dtype()}; in Types()
725 *reinterpret_cast<typename TypeDescriptor<OutType>::T*>(o) = Functor()(x); in Call()
732 template <typename InType, typename OutType, typename OutType2,
736 return {TypeDescriptor<InType>::Dtype(), TypeDescriptor<OutType>::Dtype(), in Types()
746 std::tie(*reinterpret_cast<typename TypeDescriptor<OutType>::T*>(o0), in Call()
756 template <typename InType, typename OutType, typename Functor>
760 TypeDescriptor<OutType>::Dtype()}; in Types()
770 *reinterpret_cast<typename TypeDescriptor<OutType>::T*>(o) = in Call()
779 template <typename InType, typename InType2, typename OutType, typename Functor>
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/risc/experimental/ |
| D | risc_shape_op.cc | 24 template <typename OutType>
|
| /external/libcxx/test/libcxx/utilities/function.objects/func.require/ |
| D | invoke_helpers.h | 68 typedef typename apply<T>::type OutType; in operator() typedef
69 return static_cast<OutType>(obj); in operator()
|
| /external/llvm-project/libcxx/test/libcxx/utilities/function.objects/func.require/ |
| D | invoke_helpers.h | 67 typedef typename apply<T>::type OutType; in operator() typedef
68 return static_cast<OutType>(obj); in operator()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass.cc | 192 ops::Shape::OutType(DT_INT64)); in ComputeSliceSize()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad.cc | 375 scope, Shape(scope, op.input(0), Shape::OutType(DT_INT64)), in FractionalAvgPoolGradHelper()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/TableGen/ |
| D | TGParser.cpp | 1394 RecTy *OutType; in ParseOperation() local
1396 OutType = InEltType; in ParseOperation()
1403 OutType = RHSt->getType()->getListTy(); in ParseOperation()
1406 return (TernOpInit::get(TernOpInit::FOREACH, LHS, MHS, RHS, OutType)) in ParseOperation()
|
| /external/llvm-project/llvm/lib/TableGen/ |
| D | TGParser.cpp | 1767 RecTy *OutType = InEltType; in ParseOperationForEachFilter() local
1774 OutType = RHSt->getType()->getListTy(); in ParseOperationForEachFilter()
1776 OutType = InEltType->getListTy(); in ParseOperationForEachFilter()
1781 LHS, MHS, RHS, OutType)) in ParseOperationForEachFilter()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 492 auto attrs = ops::Shape::OutType(DT_INT64); in TEST_F()
612 auto attrs = ops::Size::OutType(DT_INT64); in TEST_F()
|