| /external/tensorflow/tensorflow/lite/kernels/ |
| D | conv3d_transpose.cc | 64 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()
|
| D | conv3d.cc | 72 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()
|
| D | exp.cc | 54 #define TF_LITE_EXP(kernel_type, data_type) \ in Eval() argument
|
| D | activations.cc | 1075 KernelType kernel_type = kGenericOptimized) { in SoftmaxFloat()
1094 KernelType kernel_type = kGenericOptimized) { in SoftmaxQuantized()
1112 KernelType kernel_type) { in SoftmaxQuantized()
1136 KernelType kernel_type) { in SoftmaxQuantized()
1160 KernelType kernel_type) { in SoftmaxQuantized()
|
| D | conv.cc | 173 KernelType kernel_type) { in IsIm2ColRequired()
225 bool is_per_channel, KernelType kernel_type, size_t im2col_bytes) { in AllocateTemporaryTensorsIfRequired()
329 TfLiteStatus Prepare(KernelType kernel_type, TfLiteContext* context, in Prepare()
|
| D | lstm.cc | 49 TfLiteLSTMKernelType kernel_type; member
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | image_grad_test.cc | 200 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
|
| D | image_grad.cc | 88 string kernel_type; in ScaleAndTranslateGradHelper() local
|
| /external/libaom/libaom/test/ |
| D | hiprec_convolve_test_util.cc | 24 InterpKernel vkernel, int kernel_type = 2) { in generate_kernels()
111 for (int kernel_type = 0; kernel_type < 3; kernel_type++) { in RunCheckOutput() local
253 for (int kernel_type = 0; kernel_type < 3; kernel_type++) { in RunCheckOutput() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | depthwiseconv_quantized_test.cc | 225 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/testing/kernel_test/ |
| D | util.h | 42 string kernel_type; member
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | scale_and_translate_op_test.cc | 61 functor::SamplingKernelType kernel_type) { in Create()
424 for (const string& kernel_type : kKernelTypes) { in TEST_F() local
|
| D | scale_and_translate_op.cc | 191 const functor::SamplingKernelType kernel_type, in ComputeSpans()
239 const functor::SamplingKernelType kernel_type, in ComputeGradSpans()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_grad_test_base.py | 379 kernel_type=kernel_type, argument
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 531 TfLiteLSTMKernelType kernel_type; member
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 453 TfLiteLSTMKernelType kernel_type; member
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | operations.h | 398 LstmKernelType kernel_type = LstmKernelType::BASIC; member
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 272 TfLiteLSTMKernelType kernel_type; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 514 llvm::FunctionType* kernel_type = llvm::FunctionType::get( in BuildKernelPrototype() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2431 Type kernel_type = in matchAndRewrite() local
|