Searched refs:target_context (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 450 InferenceContext* target_context, const Node* node, int dst_idx, in ConstantPartialShape() argument
476 *result = target_context->UnknownShape(); in ConstantPartialShape()
479 *result = target_context->UnknownShape(); in ConstantPartialShape()
495 *result = target_context->Scalar(); in ConstantPartialShape()
505 target_context->MakeShapeFromTensor(&t, src_shape, result).ok()) { in ConstantPartialShape()
512 if (!ConstantPartialShape(target_context, input_edge->src(), 0, in ConstantPartialShape()
516 target_context->MakeShapeFromTensor(nullptr, src_shape, result)); in ConstantPartialShape()
518 if (!target_context->RankKnown(pre_cast_shape)) { in ConstantPartialShape()
520 *result = target_context->UnknownShape(); in ConstantPartialShape()
527 *result = target_context->MakeShape(std::vector<DimensionHandle>( in ConstantPartialShape()
[all …]
|
| D | shape_refiner.h | 265 Status ConstantPartialShape(shape_inference::InferenceContext* target_context,
|
| /external/tensorflow/tensorflow/python/checkpoint/ |
| D | save_util_v1.py | 297 target_context = to_graph.as_default
299 target_context = ops.NullContextmanager
300 with target_context():
|