Searched refs:xla_op_context (Results 1 – 1 of 1) sorted by relevance
231 XlaOpKernelContext xla_op_context(op_context); in CompileFunctionalNode() local236 XlaCompiler* compiler = xla_op_context.compiler(); in CompileFunctionalNode()256 TF_RETURN_IF_ERROR(PrepareArguments(&xla_op_context, graph.get(), expressions, in CompileFunctionalNode()303 xla_op_context.SetConstantOutput(i, result.outputs[i].constant_value); in CompileFunctionalNode()306 xla_op_context.SetTensorListOutput( in CompileFunctionalNode()309 xla_op_context.SetOutput( in CompileFunctionalNode()