Searched refs:init_data_size (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | softmax_test.cc | 54 size_t init_data_size = 0; in TestSoftmaxFloat() local
57 user_data = registration->init(&context, init_data, init_data_size); in TestSoftmaxFloat()
120 size_t init_data_size = 0; in TestSoftmaxQuantized() local
123 user_data = registration->init(&context, init_data, init_data_size); in TestSoftmaxQuantized()
|
| D | depthwise_conv_test.cc | 73 size_t init_data_size = 0; in TestDepthwiseConvFloat() local
76 user_data = registration->init(&context, init_data, init_data_size); in TestDepthwiseConvFloat()
153 size_t init_data_size = 0; in TestDepthwiseConvQuantized() local
156 user_data = registration->init(&context, init_data, init_data_size); in TestDepthwiseConvQuantized()
|
| D | fully_connected_test.cc | 66 size_t init_data_size = 0; in TestFullyConnectedFloat() local
69 user_data = registration->init(&context, init_data, init_data_size); in TestFullyConnectedFloat()
144 size_t init_data_size = 0; in TestFullyConnectedQuantized() local
147 user_data = registration->init(&context, init_data, init_data_size); in TestFullyConnectedQuantized()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| D | micro_interpreter.cc | 206 size_t init_data_size; in Invoke() local
209 init_data_size = custom_data_size; in Invoke()
212 init_data_size = 0; in Invoke()
216 user_data = registration->init(&context_, init_data, init_data_size); in Invoke()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.cc | 121 const char* init_data, size_t init_data_size, void* builtin_data, in AddNodeWithParameters() argument
124 init_data_size, builtin_data, in AddNodeWithParameters()
|
| D | interpreter.h | 149 size_t init_data_size, void* builtin_data,
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 520 const char* init_data, size_t init_data_size, void* builtin_data, in AddNodeWithParameters() argument
553 node.user_data = OpInit(*registration, init_data, init_data_size); in AddNodeWithParameters()
568 node.custom_initial_data_size = init_data_size; in AddNodeWithParameters()
|
| D | subgraph.h | 74 size_t init_data_size, void* builtin_data,
|