Searched defs:input_offsets (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| H A D | depthwise_conv_hybrid.h | 49 const int32_t* input_offsets, int thread_start, int thread_end, in DepthwiseConvHybridGeneral()
267 const float* per_channel_scales, const int32_t* input_offsets, in DepthwiseConvHybridWithRounding()
326 const float* per_channel_scales, const int32_t* input_offsets, in DepthwiseConvHybridImpl()
388 const int32_t* input_offsets; member
400 const float* per_channel_scales, int32_t* input_offsets, in DepthwiseConvHybridPerChannel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | fully_connected.cc | 391 TfLiteTensor* input_offsets; in PrepareImpl() local
517 TfLiteTensor* input_offsets, TfLiteTensor* output) { in EvalHybridDense()
586 TfLiteTensor* input_offsets, TfLiteTensor* output) { in EvalSparseHybridImpl()
713 TfLiteTensor* input_offsets; member
723 TfLiteTensor* input_offsets, TfLiteTensor* output) { in EvalHybrid()
966 TfLiteTensor* input_offsets; in EvalQuantized() local
|
| H A D | batch_matmul.cc | 265 TfLiteTensor* input_offsets; in InitializeTemporaries() local
453 TfLiteTensor* input_offsets, TfLiteTensor* output) { in EvalHybrid()
634 TfLiteTensor* input_offsets; in EvalQuantized() local
|
| H A D | dequantize_test.cc | 128 std::vector<int64_t> input_offsets(zero_points); in DequantizePerChannelOpModel() local
|
| H A D | depthwise_conv.cc | 264 TfLiteTensor* input_offsets; in Prepare() local
|
| H A D | conv.cc | 610 TfLiteTensor* input_offsets; in Prepare() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
| H A D | NestedTensorBinaryOps.cu | 58 const int64_t* input_offsets, // [batch_size] in nested_op_dense_kernelLauncher()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | tensor_utils_test.cc | 427 const std::vector<int32_t> input_offsets = {1, 1, 1, 1}; in TEST() local
1056 std::vector<int32_t> input_offsets; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | im2col_utils.h | 250 const int32_t* input_offsets, const int input_offsets_size, in Im2col()
|