Searched refs:node_context (Results 1 – 1 of 1) sorted by relevance
69 InferenceContext* node_context = CHECK_NOTNULL(refiner->GetContext(node)); in InferShapesForFunctionSubNode() local85 node_context->set_output(0, outer_context->input(index)); in InferShapesForFunctionSubNode()89 node_context->set_output_handle_shapes_and_types(0, *resource); in InferShapesForFunctionSubNode()109 node_context->ShapeHandleToProto(node_context->input(0), &proto); in InferShapesForFunctionSubNode()113 auto* resource = node_context->input_handle_shapes_and_types(0); in InferShapesForFunctionSubNode()319 InferenceContext* node_context = node_ext_context->get_context(); in UpdateNode() local322 TF_RETURN_IF_ERROR(node_context->construction_status()); in UpdateNode()342 ShapeHandle existing_input = node_context->input(dst_input); in UpdateNode()344 if (node_context->MergeInput(dst_input, c->output(src_output))) { in UpdateNode()345 if (!SameDefinedShape(node_context, node_context->input(dst_input), in UpdateNode()[all …]