Searched refs:custom_initial_data (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 217 tensorflow::Status InitializeNodeDef(const void* custom_initial_data, in InitializeNodeDef() argument
219 if (!custom_initial_data) { in InitializeNodeDef()
227 reinterpret_cast<const uint8_t*>(custom_initial_data), in InitializeNodeDef()
412 status = node_data.InitializeNodeDef(node->custom_initial_data, in Init()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 474 const void* custom_initial_data; member
559 const void* custom_initial_data; member
|
| /external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/ |
| D | pooling_slicing_test.cc | 89 node.custom_initial_data = nullptr; in TestAveragePoolingQuantized()
164 node.custom_initial_data = nullptr; in TestMaxPoolQuantized()
|
| D | fully_connected_slicing_test.cc | 101 node.custom_initial_data = nullptr; in TestFullyConnectedQuantized()
|
| D | depthwise_conv_slicing_test.cc | 94 node.custom_initial_data = nullptr; in ValidateDepthwiseConvGoldens()
|
| D | conv_slicing_test.cc | 167 node.custom_initial_data = nullptr; in ValidateConvGoldens()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ethos_u/ |
| D | ethosu.cc | 81 static_cast<uint8_t const*>(node->custom_initial_data); in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/perception/ |
| D | max_unpooling_2d.cc | 57 reinterpret_cast<const TfLitePoolParams*>(node->custom_initial_data); in Prepare()
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_interpreter.cc | 209 init_data = reinterpret_cast<const char*>(node->custom_initial_data); in AllocateTensors()
|
| D | micro_allocator.cc | 859 node->custom_initial_data = custom_data; in PrepareNodeAndRegistrationDataFromFlatbuffer()
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 699 TF_LITE_ENSURE(context, node->custom_initial_data); in PrepareParseExample()
705 reinterpret_cast<const uint8_t*>(node->custom_initial_data), in PrepareParseExample()
726 reinterpret_cast<const uint8_t*>(node->custom_initial_data), in PrepareParseExample()
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib.cc | 137 reinterpret_cast<const uint8_t*>(node.custom_initial_data), in ExportOperators()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 794 node.custom_initial_data = init_data; in AddNodeWithParameters()
797 node.custom_initial_data = nullptr; in AddNodeWithParameters()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 1009 std::memcpy(&deconv_params, node->custom_initial_data, in VisitNode()
1018 std::memcpy(&pool_params, node->custom_initial_data, in VisitNode()
1026 std::memcpy(&pool_params, node->custom_initial_data, in VisitNode()
|