Home
last modified time | relevance | path

Searched refs:KernelType (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dpooling.cc38 enum KernelType { enum
122 template <KernelType kernel_type>
150 template <KernelType kernel_type>
201 template <KernelType kernel_type>
228 template <KernelType kernel_type>
257 template <KernelType kernel_type>
282 template <KernelType kernel_type>
311 template <KernelType kernel_type>
337 template <KernelType kernel_type>
364 template <KernelType kernel_type>
Ddiv.cc30 enum KernelType { enum
80 template <KernelType kernel_type>
127 template <KernelType kernel_type>
Dfloor.cc30 enum KernelType { enum
46 template <KernelType type>
Dexp.cc30 enum KernelType { enum
53 template <KernelType kernel_type>
Dfake_quant.cc30 enum KernelType { enum
64 template <KernelType kernel_type>
Dmul.cc31 enum KernelType { enum
108 template <KernelType kernel_type>
156 template <KernelType kernel_type>
244 template <KernelType kernel_type>
Dlocal_response_norm.cc29 enum KernelType { enum
58 template <KernelType kernel_type>
Dadd.cc31 enum KernelType { enum
175 template <KernelType kernel_type>
222 template <KernelType kernel_type>
298 template <KernelType kernel_type>
Ddepthwise_conv.cc46 enum KernelType { enum
188 template <KernelType kernel_type>
224 template <KernelType kernel_type>
294 template <KernelType kernel_type>
Dconv.cc44 enum KernelType { enum
419 template <KernelType kernel_type>
431 KernelType effective_kernel_type; in EvalQuantized()
521 template <KernelType kernel_type>
529 KernelType effective_kernel_type; in EvalFloat()
588 template <KernelType kernel_type>
661 template <KernelType kernel_type>
Dtranspose.cc30 enum KernelType { enum
86 template <KernelType kernel_type>
Dspace_to_depth.cc30 enum KernelType { enum
75 template <KernelType kernel_type>
Dmaximum_minimum.cc30 enum KernelType { enum
99 template <KernelType kernel_type, typename OpType>
Dl2norm.cc30 enum KernelType { enum
72 template <KernelType kernel_type>
Dfully_connected.cc42 enum KernelType { enum
312 template <KernelType kernel_type>
386 template <KernelType kernel_type>
429 template <KernelType kernel_type>
462 template <KernelType kernel_type>
Dsub.cc32 enum KernelType { enum
225 template <KernelType kernel_type>
272 template <KernelType kernel_type>
342 template <KernelType kernel_type>
Dresize_bilinear.cc29 enum KernelType { enum
76 template <KernelType kernel_type>
Dmfcc.cc32 enum KernelType { enum
93 template <KernelType kernel_type>
Dresize_nearest_neighbor.cc29 enum KernelType { enum
74 template <KernelType kernel_type>
Daudio_spectrogram.cc35 enum KernelType { enum
100 template <KernelType kernel_type>
Dbatch_to_space_nd.cc31 enum KernelType { enum
118 template <KernelType kernel_type>
Dspace_to_batch_nd.cc31 enum KernelType { enum
107 template <KernelType kernel_type>
Dconcatenation.cc36 enum KernelType { enum
103 template <KernelType kernel_type>
/external/tensorflow/tensorflow/core/kernels/
Dsampling_kernels_test.cc26 template <typename KernelType>
27 void TestKernelValues(const KernelType& kernel, const std::vector<float>& x, in TestKernelValues()
Dscale_and_translate_op_test.cc47 template <typename KernelType>
50 explicit TypedDynamicKernel(const KernelType& kernel) : kernel_(kernel) {} in TypedDynamicKernel()
53 const KernelType kernel_;
56 template <typename KernelType>
57 std::unique_ptr<const DynamicKernel> CreateKernel(const KernelType& kernel) { in CreateKernel()
58 return MakeUnique<TypedDynamicKernel<KernelType>>(kernel); in CreateKernel()

12