Searched refs:xla_context (Results 1 – 12 of 12) sorted by relevance
138 xla_context = control_flow_ops.XLAControlFlowContext()139 xla_context.Enter()157 xla_context.Exit()172 xla_context = control_flow_ops.XLAControlFlowContext()173 xla_context.Enter()219 xla_context.Exit()233 xla_context = control_flow_ops.XLAControlFlowContext()234 xla_context.Enter()240 xla_context.Exit()
39 XlaContext* XlaOpKernelContext::xla_context() const { in xla_context() function in tensorflow::XlaOpKernelContext44 return xla_context()->builder(); in builder()48 return xla_context()->compiler(); in compiler()574 return xla_context()->GetOrCreateMax(type); in GetOrCreateMax()579 return xla_context()->GetOrCreateMin(type); in GetOrCreateMin()584 return xla_context()->GetOrCreateAdd(type); in GetOrCreateAdd()589 return xla_context()->GetOrCreateMul(type); in GetOrCreateMul()601 XlaOpKernelContext xla_context(context); in Compute() local602 Compile(&xla_context); in Compute()
63 XlaContext* xla_context() const;
176 "xla_context.cc",191 "xla_context.h",
115 Status ExecuteGraph(XlaContext* xla_context, std::unique_ptr<Graph> graph, in ExecuteGraph() argument125 xla_context->Ref(); in ExecuteGraph()133 xla_context)); in ExecuteGraph()
42 for (const auto& resource : context->xla_context()->resources()) { in Compile()
54 ctx->xla_context()->SetRetval(index_, ctx->InputExpression(0)); in Compile()
53 const XlaExpression& arg = ctx->xla_context()->args()[index_]; in Compile()
111 ctx->xla_context()->AddResource(XlaResource::CreateStack( in Compile()
170 ctx->xla_context()->AddResource(XlaResource::CreateTensorArray( in Compile()
699 xla_context = control_flow_ops.XLAControlFlowContext()700 xla_context.Enter()702 xla_context.Exit()736 xla_context = control_flow_ops.XLAControlFlowContext()737 xla_context.Enter()739 xla_context.Exit()
1388 xla_context = control_flow_ops.XLAControlFlowContext()1389 xla_context.Enter()1394 xla_context.Exit()1425 xla_context = control_flow_ops.XLAControlFlowContext()1426 xla_context.Enter()1429 xla_context.Exit()1467 xla_context = control_flow_ops.XLAControlFlowContext()1468 xla_context.Enter()1472 xla_context.Exit()1474 xla_context = control_flow_ops.XLAControlFlowContext()[all …]