Searched refs:node_ctx (Results 1 – 1 of 1) sorted by relevance
1071 NodeContext& node_ctx = node_to_context_[node]; in AddNode() local1072 TF_RETURN_IF_ERROR(function_library_.LookUp(node->op(), &node_ctx.op_data)); in AddNode()1074 if (node_ctx.op_data->is_function_op) { in AddNode()1078 TF_RETURN_IF_ERROR(InOutTypesForNode(*node, node_ctx.op_data->op_def, in AddNode()1079 &node_ctx.input_types, in AddNode()1080 &node_ctx.output_types)); in AddNode()1083 const int num_inputs = node_ctx.input_types.size(); in AddNode()1090 node_ctx.inference_context.reset(new InferenceContext( in AddNode()1091 graph_def_version_, node, node_ctx.op_data->op_def, input_shapes, in AddNode()1094 const Status s = node_ctx.inference_context->construction_status(); in AddNode()[all …]