Searched refs:src_context (Results 1 – 1 of 1) sorted by relevance
457 InferenceContext* src_context = GetContext(input_edge->src()); in ConstantPartialShape() local458 if (src_context == nullptr) return errors::Internal("Missing src context"); in ConstantPartialShape()459 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape()461 if (src_context->Value(src_context->Rank(src_shape)) == 0) { in ConstantPartialShape()486 TF_RETURN_IF_ERROR(src_context->WithRank(src_shape, 1, &src_shape)); in ConstantPartialShape()489 if (src_context->Value(src_context->Dim(src_shape, 0)) == 0) { in ConstantPartialShape()494 *result = src_context->input(0); in ConstantPartialShape()496 *result = src_context->input(input_edge->src_output()); in ConstantPartialShape()500 for (int i = 0; i < src_context->num_inputs(); ++i) { in ConstantPartialShape()517 src_op == "Concat" ? 0 : src_context->num_inputs() - 1; in ConstantPartialShape()[all …]