Searched refs:node_ctx (Results 1 – 1 of 1) sorted by relevance
1315 NodeContext& node_ctx = node_to_context_[node]; in AddNode() local1321 function_library_.LookUp(function.name(), &node_ctx.op_data)); in AddNode()1323 if (node_ctx.op_data->is_function_op) { in AddNode()1327 TF_RETURN_IF_ERROR(InOutTypesForNode(*node, node_ctx.op_data->op_def, in AddNode()1328 &node_ctx.input_types, in AddNode()1329 &node_ctx.output_types)); in AddNode()1332 const int num_inputs = node_ctx.input_types.size(); in AddNode()1339 node_ctx.inference_context.reset(new InferenceContext( in AddNode()1340 graph_def_version_, *node, node_ctx.op_data->op_def, input_shapes, in AddNode()1343 const Status s = node_ctx.inference_context->construction_status(); in AddNode()[all …]