Searched defs:kernel_type (Results 1 – 13 of 13) sorted by relevance
| /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/tensorflow/tensorflow/lite/testing/kernel_test/ |
| D | util.h | 42 string kernel_type; member
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | exp.cc | 57 #define TF_LITE_EXP(kernel_type, data_type) \ in Eval() argument
|
| D | slice.cc | 175 #define TF_LITE_SLICE(data_type, kernel_type) \ in Eval() argument
|
| D | strided_slice.cc | 210 #define TF_LITE_STRIDED_SLICE(kernel_type, data_type) \ in Eval() argument
|
| D | reduce.cc | 331 #define TF_LITE_MEAN(kernel_type, data_type, temp_data_type) \ in EvalMean() argument
|
| D | lstm.cc | 45 TfLiteLSTMKernelType kernel_type; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | depthwiseconv_quantized_test.cc | 150 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConv() local
172 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConv() local
204 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConv() local
222 DotProduct3x3KernelType kernel_type = in DispatchDepthwiseConv() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scale_and_translate_op_test.cc | 62 functor::SamplingKernelType kernel_type) { in Create()
425 for (const string& kernel_type : kKernelTypes) { in TEST_F() local
|
| D | scale_and_translate_op.cc | 189 const functor::SamplingKernelType kernel_type, in ComputeSpans()
237 const functor::SamplingKernelType kernel_type, in ComputeGradSpans()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 203 TfLiteLSTMKernelType kernel_type; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 199 llvm::FunctionType* kernel_type = llvm::FunctionType::get( in BuildKernelPrototype() local
|