/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 | 34 template <typename T, typename OutType> 37 const T* values, OutType* outputs) { in UpperBoundKernel() 41 outputs[work_unit_id] = cuda_helper::upper_bound<T, OutType>( in UpperBoundKernel() 46 template <typename T, typename OutType> 49 const T* values, OutType* outputs) { in LowerBoundKernel() 53 outputs[work_unit_id] = cuda_helper::lower_bound<T, OutType>( in LowerBoundKernel() 60 template <typename T, typename OutType> 61 struct UpperBoundFunctor<GPUDevice, T, OutType> { 66 typename TTypes<OutType, 1>::Tensor* output) { in Compute() 72 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 | 39 template <typename OutType> 50 auto vec = out->vec<OutType>(); in Compute() 59 vec(i) = static_cast<OutType>(dim_size); in Compute() 66 template <typename OutType> 78 auto vec = out->vec<OutType>(); in Compute() 88 vec(j) = static_cast<OutType>(dim_size); in Compute() 112 template <typename OutType> 129 out->scalar<OutType>()() = static_cast<OutType>(size); in Compute()
|
D | sparse_cross_op.cc | 139 template <typename OutType> 149 const OutType& cross) const { in Update() 156 auto value_vec = values_out_->vec<OutType>(); in Update()
|
/external/tensorflow/tensorflow/core/util/ |
D | cuda_kernel_helper.h | 96 template <typename T, typename OutType = int32> 97 __device__ OutType upper_bound(const T* first, OutType count, T val) { in upper_bound() 100 OutType step = 0; in upper_bound() 116 template <typename T, typename OutType = int32> 117 __device__ OutType lower_bound(const T* first, OutType count, T val) { in lower_bound() 120 OutType step = 0; in lower_bound()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | shape.cc | 29 template <typename OutType> 30 void ExtractShape(const TfLiteTensor* input, OutType* output_data) { in ExtractShape()
|
/external/tensorflow/tensorflow/contrib/layers/kernels/ |
D | sparse_feature_cross_kernel.cc | 139 template <typename OutType> 149 const OutType& cross) const { in Update() 156 auto value_vec = values_out_->vec<OutType>(); in Update()
|
/external/tensorflow/tensorflow/python/lib/core/ |
D | bfloat16.cc | 492 template <typename InType, typename OutType, typename Functor> 501 *reinterpret_cast<OutType*>(o) = Functor()(x, y); in BinaryUFunc()
|
/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/tensorflow/tensorflow/compiler/jit/ |
D | increase_dynamism_for_auto_jit_pass.cc | 192 ops::Shape::OutType(DT_INT64)); in ComputeSliceSize()
|
/external/swiftshader/third_party/llvm-7.0/llvm/lib/TableGen/ |
D | TGParser.cpp | 1291 RecTy *OutType; in ParseOperation() local 1293 OutType = InEltType; in ParseOperation() 1300 OutType = RHSt->getType()->getListTy(); in ParseOperation() 1303 return (TernOpInit::get(TernOpInit::FOREACH, LHS, MHS, RHS, OutType)) in ParseOperation()
|
/external/tensorflow/tensorflow/cc/gradients/ |
D | nn_grad.cc | 375 scope, Shape(scope, op.input(0), Shape::OutType(DT_INT64)), in FractionalAvgPoolGradHelper()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | shape_refiner_test.cc | 507 auto attrs = ops::Shape::OutType(DT_INT64); in TEST_F() 627 auto attrs = ops::Size::OutType(DT_INT64); in TEST_F()
|