Searched refs:xla_op_context (Results 1 – 1 of 1) sorted by relevance
224 XlaOpKernelContext xla_op_context(op_context); in CompileFunctionalNode() local229 XlaCompiler* compiler = xla_op_context.compiler(); in CompileFunctionalNode()250 PrepareArguments(&xla_op_context, graph.get(), expressions, &arguments)); in CompileFunctionalNode()292 xla_op_context.SetConstantOutput(i, result.outputs[i].constant_value); in CompileFunctionalNode()294 xla_op_context.SetOutput( in CompileFunctionalNode()