Searched refs:XlaContext (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.cc | 39 const char XlaContext::kXlaContextResourceName[] = "_xla_context";
43 /* static */ XlaContext& XlaContext::Get(const OpKernelContext* ctx) { in Get()
47 XlaContext* context; in Get()
57 void XlaContext::set_args(std::vector<XlaExpression> args) { in set_args()
61 XlaContext::XlaContext(XlaCompiler* compiler, xla::XlaBuilder* builder) in XlaContext() function in tensorflow::XlaContext
64 string XlaContext::DebugString() const { return "XLA JIT context"; } in DebugString()
66 void XlaContext::SetRetval(int index, const XlaExpression& expression) { in SetRetval()
73 XlaResource* XlaContext::AddResource(std::unique_ptr<XlaResource> resource) { in AddResource()
78 const xla::XlaComputation* XlaContext::GetOrCreateMax(const DataType type) { in GetOrCreateMax()
94 const xla::XlaComputation* XlaContext::GetOrCreateMin(const DataType type) { in GetOrCreateMin()
[all …]
|
| D | xla_context.h | 40 class XlaContext : public ResourceBase {
43 static XlaContext& Get(const OpKernelContext* ctx);
47 XlaContext(XlaCompiler* compiler, xla::XlaBuilder* builder);
138 TF_DISALLOW_COPY_AND_ASSIGN(XlaContext);
|
| D | xla_compiler.h | 44 class XlaContext; variable
433 XlaContext* context,
|
| D | graph_compiler.cc | 226 XlaContext& context = XlaContext::Get(op_context); in CompileFunctionalNode()
|
| D | xla_compilation_device.cc | 94 auto* b = XlaContext::Get(context).builder(); in Compute()
|
| D | xla_compiler.cc | 115 Status ExecuteGraph(XlaContext* xla_context, std::unique_ptr<Graph> graph, in ExecuteGraph()
132 step_container->name(), XlaContext::kXlaContextResourceName, in ExecuteGraph()
704 bool use_tuple_arg, xla::XlaBuilder* builder, XlaContext* context, in BuildArguments()
1069 XlaContext* context = new XlaContext(this, &builder); in CompileGraph()
|
| D | xla_op_kernel.h | 63 XlaContext* xla_context() const;
|
| D | xla_op_kernel.cc | 39 XlaContext* XlaOpKernelContext::xla_context() const { in xla_context()
40 return &XlaContext::Get(context_); in xla_context()
|