Home
last modified time | relevance | path

Searched refs:custom_initial_data_size (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/ethos_u/
Dethosu.cc44 TF_LITE_ENSURE(context, node->custom_initial_data_size > 0); in Prepare()
82 auto root = flexbuffers::GetRoot(custom_data, node->custom_initial_data_size); in Eval()
/external/tensorflow/tensorflow/lite/c/
Dcommon.h475 int custom_initial_data_size; member
560 int custom_initial_data_size; member
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dpooling_slicing_test.cc90 node.custom_initial_data_size = 0; in TestAveragePoolingQuantized()
165 node.custom_initial_data_size = 0; in TestMaxPoolQuantized()
Dfully_connected_slicing_test.cc102 node.custom_initial_data_size = 0; in TestFullyConnectedQuantized()
Ddepthwise_conv_slicing_test.cc95 node.custom_initial_data_size = 0; in ValidateDepthwiseConvGoldens()
Dconv_slicing_test.cc168 node.custom_initial_data_size = 0; in ValidateConvGoldens()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc218 int custom_initial_data_size) { in InitializeNodeDef() argument
228 custom_initial_data_size) in InitializeNodeDef()
413 node->custom_initial_data_size); in Init()
/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter.cc210 init_data_size = node->custom_initial_data_size; in AllocateTensors()
Dmicro_allocator.cc860 node->custom_initial_data_size = custom_data_size; in PrepareNodeAndRegistrationDataFromFlatbuffer()
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib.cc138 node.custom_initial_data_size); in ExportOperators()
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example.cc706 node->custom_initial_data_size) in PrepareParseExample()
727 node->custom_initial_data_size) in PrepareParseExample()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc795 node.custom_initial_data_size = init_data_size; in AddNodeWithParameters()
798 node.custom_initial_data_size = 0; in AddNodeWithParameters()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc1010 node->custom_initial_data_size); in VisitNode()
1019 node->custom_initial_data_size); in VisitNode()
1027 node->custom_initial_data_size); in VisitNode()