Home
last modified time | relevance | path

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

/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.cc88 string kernel_type; in ScaleAndTranslateGradHelper() local
/external/libaom/libaom/test/
Dhiprec_convolve_test_util.cc24 InterpKernel vkernel, int kernel_type = 2) { in generate_kernels()
111 for (int kernel_type = 0; kernel_type < 3; kernel_type++) { in RunCheckOutput() local
255 for (int kernel_type = 0; kernel_type < 3; kernel_type++) { in RunCheckOutput() local
/external/tensorflow/tensorflow/lite/testing/kernel_test/
Dutil.h42 string kernel_type; member
/external/tensorflow/tensorflow/lite/kernels/
Dexp.cc54 #define TF_LITE_EXP(kernel_type, data_type) \ in Eval() argument
Dconv.cc166 KernelType kernel_type) { in IsIm2ColRequired()
220 KernelType kernel_type) { in AllocateTemporaryTensorsIfRequired()
312 TfLiteStatus Prepare(KernelType kernel_type, TfLiteContext* context, in Prepare()
Dlstm.cc49 TfLiteLSTMKernelType 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/lite/tools/versioning/
Dop_version.h52 LSTMKernelType kernel_type; member
/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.cc190 const functor::SamplingKernelType kernel_type, in ComputeSpans()
238 const functor::SamplingKernelType kernel_type, in ComputeGradSpans()
/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test_base.py377 kernel_type=kernel_type, argument
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc520 TfLiteLSTMKernelType kernel_type; member
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc452 TfLiteLSTMKernelType kernel_type; member
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h389 LstmKernelType kernel_type = LstmKernelType::BASIC; member
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h271 TfLiteLSTMKernelType kernel_type; member
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc568 llvm::FunctionType* kernel_type = llvm::FunctionType::get( in BuildKernelPrototype() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1824 Type kernel_type = in matchAndRewrite() local