| /external/tensorflow/tensorflow/compiler/tests/ |
| D | cond_test.py | 44 xla_context = control_flow_ops.XLAControlFlowContext()
45 xla_context.Enter()
59 xla_context.Exit()
64 xla_context = control_flow_ops.XLAControlFlowContext()
65 xla_context.Enter()
79 xla_context.Exit()
85 xla_context = control_flow_ops.XLAControlFlowContext()
86 xla_context.Enter()
99 xla_context.Exit()
103 xla_context = control_flow_ops.XLAControlFlowContext()
[all …]
|
| D | while_test.py | 139 xla_context = control_flow_ops.XLAControlFlowContext()
140 xla_context.Enter()
158 xla_context.Exit()
173 xla_context = control_flow_ops.XLAControlFlowContext()
174 xla_context.Enter()
220 xla_context.Exit()
234 xla_context = control_flow_ops.XLAControlFlowContext()
235 xla_context.Enter()
241 xla_context.Exit()
248 xla_context = control_flow_ops.XLAControlFlowContext()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 105 XlaContext& xla_context = XlaContext::Get(context); in Compute() local
106 auto* b = xla_context.builder(); in Compute()
110 AttachLocationToMetadata(metadata, op_kernel, xla_context); in Compute()
|
| D | xla_op_kernel.cc | 40 XlaContext* XlaOpKernelContext::xla_context() const { in xla_context() function in tensorflow::XlaOpKernelContext
45 return xla_context()->builder(); in builder()
49 return xla_context()->compiler(); in compiler()
693 return xla_context()->GetOrCreateMax(type); in GetOrCreateMax()
698 return xla_context()->GetOrCreateMin(type); in GetOrCreateMin()
703 return xla_context()->GetOrCreateAdd(type); in GetOrCreateAdd()
708 return xla_context()->GetOrCreateMul(type); in GetOrCreateMul()
720 XlaOpKernelContext xla_context(context); in Compute() local
721 Compile(&xla_context); in Compute()
726 xla_context()->StackTraceForNodeName(op_kernel().name())) { in StackTrace()
|
| D | BUILD | 348 ":xla_context",
399 ":xla_context",
423 name = "xla_context",
425 "xla_context.cc",
428 "xla_context.h",
461 ":xla_context",
|
| D | xla_op_kernel.h | 66 XlaContext* xla_context() const;
|
| D | xla_compiler.cc | 122 Status ExecuteGraph(XlaContext* xla_context, std::unique_ptr<Graph> graph, in ExecuteGraph() argument
132 xla_context->Ref(); in ExecuteGraph()
140 xla_context)); in ExecuteGraph()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | retval_op.cc | 56 ctx->xla_context()->SetRetval(index_, ctx->InputExpression(0)); in Compile()
|
| D | arg_op.cc | 60 const XlaExpression& arg = ctx->xla_context()->args()[index_]; in Compile()
|
| D | BUILD | 155 "//tensorflow/compiler/tf2xla:xla_context",
318 "//tensorflow/compiler/tf2xla:xla_context",
338 "//tensorflow/compiler/tf2xla:xla_context",
|
| D | stack_ops.cc | 111 ctx->xla_context()->AddResource(XlaResource::CreateStack( in Compile()
|
| D | tensor_array_ops.cc | 170 ctx->xla_context()->AddResource(XlaResource::CreateTensorArray( in Compile()
|
| /external/tensorflow/tensorflow/compiler/aot/ |
| D | aot_only_var_handle_op.cc | 45 for (const auto& resource : context->xla_context()->resources()) { in Compile()
|
| D | BUILD | 312 "//tensorflow/compiler/tf2xla:xla_context",
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | BUILD | 26 "//tensorflow/compiler/tf2xla:xla_context",
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cond_v2_test.py | 1026 xla_context = control_flow_ops.XLAControlFlowContext()
1027 xla_context.Enter()
1029 xla_context.Exit()
1070 xla_context = control_flow_ops.XLAControlFlowContext()
1071 xla_context.Enter()
1073 xla_context.Exit()
1126 xla_context = control_flow_ops.XLAControlFlowContext()
1127 xla_context.Enter()
1129 xla_context.Exit()
|
| D | control_flow_ops_py_test.py | 1637 xla_context = control_flow_ops.XLAControlFlowContext()
1638 xla_context.Enter()
1650 xla_context.Exit()
1667 xla_context = control_flow_ops.XLAControlFlowContext()
1668 xla_context.Enter()
1673 xla_context.Exit()
1704 xla_context = control_flow_ops.XLAControlFlowContext()
1705 xla_context.Enter()
1708 xla_context.Exit()
1746 xla_context = control_flow_ops.XLAControlFlowContext()
[all …]
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | def_function.py | 605 xla_context = control_flow_ops.XLAControlFlowContext()
607 xla_context.Enter()
610 xla_context.Exit()
|
| D | function.py | 3185 xla_context = _enclosing_xla_context()
3186 if xla_context is not None and \
3187 xla_context.RequiresUniqueFunctionRetracing():
3188 xla_context_id = id(xla_context)
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | BUILD | 153 "//tensorflow/compiler/tf2xla:xla_context",
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | BUILD | 422 "//tensorflow/compiler/tf2xla:xla_context",
|