Searched refs:custom_initial_data (Results 1 – 8 of 8) sorted by relevance
203 tensorflow::Status InitializeNodeDef(const void* custom_initial_data, in InitializeNodeDef() argument205 if (!custom_initial_data) { in InitializeNodeDef()213 reinterpret_cast<const uint8_t*>(custom_initial_data), in InitializeNodeDef()410 status = node_data.InitializeNodeDef(node->custom_initial_data, in Init()
72 node.custom_initial_data = nullptr; in TestSoftmaxFloat()139 node.custom_initial_data = nullptr; in TestSoftmaxQuantized()
91 node.custom_initial_data = nullptr; in TestDepthwiseConvFloat()172 node.custom_initial_data = nullptr; in TestDepthwiseConvQuantized()
84 node.custom_initial_data = nullptr; in TestFullyConnectedFloat()163 node.custom_initial_data = nullptr; in TestFullyConnectedQuantized()
384 const void* custom_initial_data; member
258 node.custom_initial_data = custom_data; in Invoke()
98 reinterpret_cast<const uint8_t*>(node.custom_initial_data), in ExportOperators()
567 node.custom_initial_data = init_data; in AddNodeWithParameters()570 node.custom_initial_data = nullptr; in AddNodeWithParameters()