Home
last modified time | relevance | path

Searched defs:kernel_type (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dconv3d_transpose.cc64 KernelType kernel_type) { in AllocateTemporaryTensorsIfRequired()
137 TfLiteStatus Prepare(KernelType kernel_type, TfLiteContext* context, in Prepare()
213 void EvalFloat(KernelType kernel_type, TfLiteContext* context, TfLiteNode* node, in EvalFloat()
254 TfLiteStatus Eval(KernelType kernel_type, TfLiteContext* context, in Eval()
Dconv3d.cc72 KernelType kernel_type, TfLiteContext* context, TfLiteNode* node, in AllocateTemporaryTensorsIfRequired()
121 TfLiteStatus Prepare(KernelType kernel_type, TfLiteContext* context, in Prepare()
242 void EvalFloat(KernelType kernel_type, TfLiteContext* context, TfLiteNode* node, in EvalFloat()
284 TfLiteStatus Eval(KernelType kernel_type, TfLiteContext* context, in Eval()
Dexp.cc54 #define TF_LITE_EXP(kernel_type, data_type) \ in Eval() argument
Dreduce.cc379 KernelType kernel_type) { in Mean()
399 KernelType kernel_type, bool compute_sum) { in QuantizedMeanOrSum()
429 KernelType kernel_type) { in EvalIntegerMean()
701 OpContext* op_context, KernelType kernel_type, in EvalType()
Dactivations.cc1107 KernelType kernel_type = kGenericOptimized) { in SoftmaxFloat()
1126 KernelType kernel_type = kGenericOptimized) { in SoftmaxQuantized()
1144 KernelType kernel_type) { in SoftmaxQuantized()
1168 KernelType kernel_type) { in SoftmaxQuantized()
1192 KernelType kernel_type) { in SoftmaxQuantized()
Dconv.cc176 KernelType kernel_type) { in IsIm2ColRequired()
229 bool is_per_channel, KernelType kernel_type, size_t im2col_bytes) { in AllocateTemporaryTensorsIfRequired()
333 TfLiteStatus Prepare(KernelType kernel_type, TfLiteContext* context, in Prepare()
/external/tensorflow/tensorflow/cc/gradients/
Dimage_grad_test.cc200 Input translation, const string& kernel_type, bool antialias, in MakeOp()
213 Input translation, const string& kernel_type, in TestScaleAndTranslate()
242 for (const std::string& kernel_type : kKernelTypes) { in TEST_F() local
274 for (const std::string& kernel_type : kKernelTypes) { in TEST_F() local
291 for (const std::string& kernel_type : kKernelTypes) { in TEST_F() local
Dimage_grad.cc94 string kernel_type; in ScaleAndTranslateGradHelper() local
/external/ComputeLibrary/src/cpu/
DICpuKernel.h55 …using kernel_type = typename std::remove_reference<decltype(Derived::get_available_kernels())>::ty… variable
/external/libaom/test/
Dhiprec_convolve_test_util.cc27 InterpKernel vkernel, int kernel_type = 2) { in generate_kernels()
150 for (int kernel_type = 0; kernel_type < 6; kernel_type++) { in RunCheckOutput() local
289 for (int kernel_type = 0; kernel_type < 6; kernel_type++) { in RunCheckOutput() local
/external/ComputeLibrary/src/core/CL/
DDefaultLWSHeuristics.cpp97 cl::NDRange get_default_lws_for_type(CLKernelType kernel_type, cl::NDRange gws) in get_default_lws_for_type()
/external/tensorflow/tensorflow/lite/testing/kernel_test/
Dutil.h42 string kernel_type; member
/external/tensorflow/tensorflow/lite/kernels/internal/
Ddepthwiseconv_quantized_test.cc225 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConvImpl() local
243 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConvImpl() local
277 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConvImpl() local
297 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConvImpl() local
429 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConvImpl() local
454 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConvImpl() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_operator.cc215 mlir::TFL::LSTMKernelType kernel_type, in ConvertTFL_LSTMKernelTypeAttrForOptionWriter()
284 mlir::TFL::LSTMKernelType kernel_type; in BuildTFL_LSTMKernelTypeAttr() local
/external/google-cloud-java/java-dataplex/proto-google-cloud-dataplex-v1/src/main/proto/google/cloud/dataplex/v1/
Danalyze.proto208 KernelType kernel_type = 1 [(google.api.field_behavior) = REQUIRED]; field
/external/googleapis/google/cloud/dataplex/v1/
Danalyze.proto208 KernelType kernel_type = 1 [(google.api.field_behavior) = REQUIRED]; field
/external/ComputeLibrary/src/runtime/CL/gemm_auto_heuristics/
DCLGEMMAutoHeuristics.cpp82 const auto kernel_type = default_heuristics->select_kernel(params); in select_default_gemm_kernel() local
/external/autotest/server/cros/faft/
Dfirmware_test.py1944 def is_kernel_changed(self, kernel_type='KERN'): argument
1959 def backup_kernel(self, suffix='.original', kernel_type='KERN'): argument
1980 def is_kernel_saved(self, kernel_type='KERN'): argument
1988 def restore_kernel(self, suffix='.original', kernel_type='KERN'): argument
/external/ComputeLibrary/src/gpu/cl/operators/
DClGemmLowpMatrixMultiplyCore.cpp53 inline bool validate_gemm_kernel(CLGEMMKernelType kernel_type) in validate_gemm_kernel()
225 inline bool is_gemm_reshaped(CLGEMMKernelType kernel_type) in is_gemm_reshaped()
/external/XNNPACK/scripts/
Dconvert-assembly-to-jit.py535 def print_generator_definition(kernel_type, fn_name, arch, minmax, prefetch=''): argument
/external/tensorflow/tensorflow/core/kernels/image/
Dscale_and_translate_op_test.cc61 functor::SamplingKernelType kernel_type) { in Create()
424 for (const string& kernel_type : kKernelTypes) { in TEST_F() local
Dscale_and_translate_op.cc191 const functor::SamplingKernelType kernel_type, in ComputeSpans()
239 const functor::SamplingKernelType kernel_type, in ComputeGradSpans()
/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test_base.py375 kernel_type=kernel_type, argument
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc531 TfLiteLSTMKernelType kernel_type; member
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc453 TfLiteLSTMKernelType kernel_type; member

12