Home
last modified time | relevance | path

Searched refs:outer_context (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc69 const Node* node, InferenceContext* outer_context) { in InferShapesForFunctionSubNode() argument
70 TF_RETURN_IF_ERROR(AddNodeInternal(node, outer_context)); in InferShapesForFunctionSubNode()
81 if (index < 0 || outer_context->num_inputs() <= index) { in InferShapesForFunctionSubNode()
84 " not in [0, ", outer_context->num_inputs(), ")."); in InferShapesForFunctionSubNode()
89 if (outer_context->input(index).SameHandle(ShapeHandle())) { in InferShapesForFunctionSubNode()
94 node_context->set_output(0, outer_context->input(index)); in InferShapesForFunctionSubNode()
97 auto* resource = outer_context->input_handle_shapes_and_types(index); in InferShapesForFunctionSubNode()
109 if (index < 0 || outer_context->num_outputs() <= index) { in InferShapesForFunctionSubNode()
112 " not in [0, ", outer_context->num_outputs(), ")."); in InferShapesForFunctionSubNode()
120 TF_RETURN_IF_ERROR(outer_context->MakeShapeFromShapeProto(proto, &handle)); in InferShapesForFunctionSubNode()
[all …]
Dshape_refiner.h185 ExtendedInferenceContext* outer_context);
191 const Node* node, shape_inference::InferenceContext* outer_context);
210 shape_inference::InferenceContext* outer_context);
223 shape_inference::InferenceContext* outer_context);
236 shape_inference::InferenceContext* outer_context);
268 shape_inference::InferenceContext* outer_context);
280 shape_inference::InferenceContext* outer_context);
291 shape_inference::InferenceContext* outer_context = nullptr);
Deval_const_tensor.cc146 InferenceContext* outer_context) { in ExtractConstantSubgraph() argument
243 if (outer_context && current_node->IsArg()) { in ExtractConstantSubgraph()
251 current_node, outer_context->num_inputs(), &index)); in ExtractConstantSubgraph()
252 const Tensor* const_tensor = outer_context->input_tensor(index); in ExtractConstantSubgraph()
260 outer_context->request_input_tensor(index); in ExtractConstantSubgraph()
355 InferenceContext* outer_context) { in EvaluateConstantTensor() argument
369 if (src->IsArg() && outer_context) { in EvaluateConstantTensor()
372 GetArgNodeIndex(src, outer_context->num_inputs(), &index)); in EvaluateConstantTensor()
373 const Tensor* const_tensor = outer_context->input_tensor(index); in EvaluateConstantTensor()
376 *result = *(outer_context->input_tensor(index)); in EvaluateConstantTensor()
[all …]
Deval_const_tensor.h66 shape_inference::InferenceContext* outer_context = nullptr);
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_state.py103 while_ctxt.outer_context, stop_ctxt=curr_ctxt)
161 outer_forward_ctxt = forward_ctxt.outer_context
259 if self._grad_context.outer_context:
260 self._grad_context.outer_context.AddInnerOp(self._grad_sync)
371 value_ctxt.outer_context.Enter()
374 value_ctxt.outer_context.Exit()
412 value_ctxt = value_ctxt.outer_context
504 forward_ctxt = forward_ctxt.outer_context
577 outer_forward_ctxt = forward_ctxt.outer_context
609 outer_forward_ctxt = forward_ctxt.outer_context
[all …]
Dcontrol_flow_util.py127 ctxt = ctxt.outer_context
187 ctxt = ctxt.outer_context
209 ctxt = ctxt.outer_context
228 ctxt = ctxt.outer_context
246 ctxt = ctxt.outer_context
254 ctxt = ctxt.outer_context
Dcontrol_flow_ops.py695 def outer_context(self): member in ControlFlowContext
1330 assert context_t.outer_context == context_f.outer_context
1331 if context_t.outer_context is None:
1709 forward_ctxt = forward_ctxt.outer_context
1986 if self.outer_context:
1987 self.outer_context.Enter()
1989 if self.outer_context:
1990 self.outer_context.Exit()
1993 if (isinstance(self.outer_context, WhileContext) and
1994 self.outer_context.grad_state is not None):
[all …]
Dspecial_math_ops.py602 context = context.outer_context
/external/tensorflow/tensorflow/core/lib/core/
Dthreadpool_test.cc42 Context outer_context(ContextKind::kThread); in TEST() local
53 pool.Schedule([&outer_context, &work, i]() { in TEST()
55 ASSERT_EQ(outer_context, inner_context); in TEST()
262 Context outer_context(ContextKind::kThread); in TEST() local
274 [&outer_context, &work](int64 begin, int64 end) { in TEST()
276 ASSERT_EQ(outer_context, inner_context); in TEST()
288 Context outer_context(ContextKind::kThread); in TEST() local
304 [&outer_context, &work](int64 begin, int64 end) { in TEST()
306 ASSERT_EQ(outer_context, inner_context); in TEST()
/external/tensorflow/tensorflow/python/keras/utils/
Dcontrol_flow_util.py69 ctxt = ctxt.outer_context
88 ctxt = ctxt.outer_context
/external/tensorflow/tensorflow/python/distribute/
Dtpu_util.py31 context_ = context_.outer_context
/external/tensorflow/tensorflow/python/framework/
Dops.py5623 outer_context = None
5632 outer_context = default_graph.as_default
5639 outer_context = stack_entry.enter_context_fn
5642 if outer_context is None:
5647 …outer_context = _default_graph_stack._GetGlobalDefaultGraph().as_default # pylint: disable=protec…
5649 if outer_context is None:
5654 return outer_context, innermost_nonempty_device_stack
5722 outer_context, innermost_nonempty_device_stack = (
5728 with outer_context(), name_scope(
5791 outer_context, _ = _get_outer_context_and_inner_device_stack()
[all …]
Dauto_control_deps.py297 switch_op._control_flow_context.outer_context)
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py207 context_ = context_.outer_context
373 graph._set_control_flow_context(self.outer_context)
833 context = context.outer_context
848 context = context.outer_context
2093 context = context.outer_context
Dtpu_embedding_v2.py1049 ctx = ctx.outer_context
Dtensor_tracer.py1556 op_control_flow_context = op_control_flow_context.outer_context
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops.py132 control_flow_context = control_flow_context.outer_context
/external/tensorflow/tensorflow/python/eager/
Dfunction.py389 context_ = context_.outer_context