Searched refs:target_context (Results 1 – 3 of 3) sorted by relevance
451 Status ShapeRefiner::ConstantPartialShape(InferenceContext* target_context, in ConstantPartialShape() argument473 *result = target_context->UnknownShape(); in ConstantPartialShape()476 *result = target_context->UnknownShape(); in ConstantPartialShape()492 *result = target_context->Scalar(); in ConstantPartialShape()506 dims.push_back(size < 0 ? target_context->UnknownDim() in ConstantPartialShape()507 : target_context->MakeDim(size)); in ConstantPartialShape()509 dims.push_back(target_context->UnknownDim()); in ConstantPartialShape()512 *result = target_context->MakeShape(dims); in ConstantPartialShape()514 *result = target_context->Scalar(); in ConstantPartialShape()523 TF_RETURN_IF_ERROR(ConstantPartialShape(target_context, input_edge->src(), in ConstantPartialShape()[all …]
251 Status ConstantPartialShape(shape_inference::InferenceContext* target_context,
387 target_context = to_graph.as_default389 target_context = ops.NullContextmanager390 with target_context():