Searched refs:XlaContext (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.cc | 38 const char XlaContext::kXlaContextResourceName[] = "_xla_context";
42 /* static */ XlaContext& XlaContext::Get(const OpKernelContext* ctx) { in Get()
46 XlaContext* context; in Get()
56 void XlaContext::set_args(std::vector<XlaExpression> args) { in set_args()
60 XlaContext::XlaContext(XlaCompiler* compiler, xla::XlaBuilder* builder, in XlaContext() function in tensorflow::XlaContext
70 string XlaContext::DebugString() const { return "XLA JIT context"; } in DebugString()
72 void XlaContext::SetRetval(int index, const XlaExpression& expression) { in SetRetval()
80 XlaResource* XlaContext::AddResource(std::unique_ptr<XlaResource> resource) { in AddResource()
85 const xla::XlaComputation* XlaContext::GetOrCreateMax(const DataType type) { in GetOrCreateMax()
101 const xla::XlaComputation* XlaContext::GetOrCreateMin(const DataType type) { in GetOrCreateMin()
[all …]
|
| D | xla_context.h | 41 class XlaContext : public ResourceBase {
44 static XlaContext& Get(const OpKernelContext* ctx);
48 XlaContext(XlaCompiler* compiler, xla::XlaBuilder* builder,
151 TF_DISALLOW_COPY_AND_ASSIGN(XlaContext);
|
| D | xla_compilation_device.cc | 91 OpKernel* op_kernel, XlaContext& context) { in AttachLocationToMetadata()
105 XlaContext& xla_context = XlaContext::Get(context); in Compute()
|
| D | xla_compiler.h | 46 class XlaContext; variable
306 XlaContext* context,
|
| D | graph_compiler.cc | 233 XlaContext& context = XlaContext::Get(op_context); in CompileFunctionalNode()
|
| 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()
139 XlaContext::kXlaContextResourceName, in ExecuteGraph()
955 bool use_tuple_arg, xla::XlaBuilder* builder, XlaContext* context, in BuildArguments()
1312 XlaContext* context = new XlaContext(this, &builder, graph.get()); in CompileGraph()
|
| D | xla_op_kernel.cc | 40 XlaContext* XlaOpKernelContext::xla_context() const { in xla_context()
41 return &XlaContext::Get(context_); in xla_context()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 344 tensorflow::XlaContext* context_; // Ref-counted.
358 context_ = new tensorflow::XlaContext(/*compiler=*/nullptr, &hlo_builder_, in PrepareParams()
380 tensorflow::XlaContext::kXlaContextResourceName, context_); in PrepareParams()
|