Home
last modified time | relevance | path

Searched refs:XlaCompiler (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc63 XlaCompiler::Options DefaultOptions() { in DefaultOptions()
64 XlaCompiler::Options options; in DefaultOptions()
71 FunctionLibraryDefinition* LocalFlibDef(XlaCompiler* compiler) { in LocalFlibDef()
174 XlaCompiler compiler(DefaultOptions()); in TEST_F()
177 XlaCompiler::CompilationResult result; in TEST_F()
179 XlaCompiler::CompileOptions(), "add", std::move(graph), in TEST_F()
197 std::vector<XlaCompiler::Argument> args(2); in TEST_F()
198 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
201 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
206 XlaCompiler compiler(DefaultOptions()); in TEST_F()
[all …]
Dxla_compiler.cc57 absl::Span<const XlaCompiler::Argument> args) { in CheckSignature()
158 const std::vector<XlaCompiler::Argument>& args, in BuildComputation()
163 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn, in BuildComputation()
167 std::vector<XlaCompiler::OutputDescription>* outputs, in BuildComputation()
168 std::vector<XlaCompiler::ResourceUpdate>* resource_updates, in BuildComputation()
193 XlaCompiler::OutputDescription& output = (*outputs)[i]; in BuildComputation()
259 const XlaCompiler::Argument& arg = args[resource->arg_num()]; in BuildComputation()
273 XlaCompiler::ResourceUpdate& update = resource_updates->back(); in BuildComputation()
342 bool XlaCompiler::Argument::operator==( in operator ==()
343 const XlaCompiler::Argument& other) const { in operator ==()
[all …]
Dgraph_compiler.cc57 std::vector<XlaCompiler::Argument>* args) { in PrepareArguments()
67 XlaCompiler::Argument& arg = (*args)[i]; in PrepareArguments()
73 arg.kind = XlaCompiler::Argument::kConstant; in PrepareArguments()
85 arg.kind = XlaCompiler::Argument::kConstant; in PrepareArguments()
88 arg.kind = XlaCompiler::Argument::kParameter; in PrepareArguments()
229 XlaCompiler* compiler = xla_op_context.compiler(); in CompileFunctionalNode()
243 std::vector<XlaCompiler::Argument> arguments; in CompileFunctionalNode()
255 XlaCompiler::CompileOptions compile_options; in CompileFunctionalNode()
258 XlaCompiler::CompilationResult result; in CompileFunctionalNode()
266 if (arguments[i].kind == XlaCompiler::Argument::kConstant) { in CompileFunctionalNode()
Dtf2xla.cc237 std::vector<XlaCompiler::Argument>* xla_args) { in CreateXlaArgs()
241 XlaCompiler::Argument arg; in CreateXlaArgs()
242 arg.kind = XlaCompiler::Argument::kParameter; in CreateXlaArgs()
263 std::vector<XlaCompiler::Argument> xla_args; in ConvertGraphToXla()
272 XlaCompiler::Argument arg; in ConvertGraphToXla()
274 arg.kind = XlaCompiler::Argument::kResource; in ConvertGraphToXla()
290 XlaCompiler::Options compiler_options; in ConvertGraphToXla()
296 XlaCompiler compiler(compiler_options); in ConvertGraphToXla()
298 XlaCompiler::CompilationResult result; in ConvertGraphToXla()
299 TF_RETURN_IF_ERROR(compiler.CompileGraph(XlaCompiler::CompileOptions(), in ConvertGraphToXla()
Dxla_context.h47 XlaContext(XlaCompiler* compiler, xla::XlaBuilder* builder);
52 XlaCompiler* compiler() const { return compiler_; } in compiler()
98 XlaCompiler* const compiler_;
Dxla_compiler.h98 class XlaCompiler {
330 explicit XlaCompiler(Options options);
332 ~XlaCompiler();
431 const std::vector<XlaCompiler::Argument>& args,
494 TF_DISALLOW_COPY_AND_ASSIGN(XlaCompiler);
Dxla_context.cc61 XlaContext::XlaContext(XlaCompiler* compiler, xla::XlaBuilder* builder) in XlaContext()
Dxla_op_kernel.h241 XlaCompiler* compiler() const;
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.h71 Status Compile(const XlaCompiler::Options& options,
73 absl::Span<const XlaCompiler::Argument> args,
74 const XlaCompiler::CompileOptions& compile_options,
76 const XlaCompiler::CompilationResult** out_compilation_result,
82 const XlaCompiler::Options& options,
83 absl::Span<const XlaCompiler::Argument> args, OpKernelContext* ctx,
84 const XlaCompiler::CompileOptions& compile_options,
85 const XlaCompiler::CompilationResult** out_compilation_result,
119 absl::Span<const XlaCompiler::Argument> args);
124 const XlaCompiler::Options& options, const NameAttrList& function,
[all …]
Dxla_compilation_cache.cc119 absl::Span<const XlaCompiler::Argument> args) { in BuildSignature()
122 for (const XlaCompiler::Argument& arg : args) { in BuildSignature()
124 case XlaCompiler::Argument::kConstant: in BuildSignature()
127 case XlaCompiler::Argument::kParameter: in BuildSignature()
128 case XlaCompiler::Argument::kResource: in BuildSignature()
141 const XlaCompiler::Options& options, in BuildExecutable()
142 const XlaCompiler::CompilationResult& result, in BuildExecutable()
168 const XlaCompiler::Options& options, const NameAttrList& function, in Compile()
169 absl::Span<const XlaCompiler::Argument> args, in Compile()
170 const XlaCompiler::CompileOptions& compile_options, in Compile()
[all …]
Dxla_compile_on_demand_op.cc52 const XlaCompiler::CompilationResult* result, in Run()
115 const XlaCompiler::CompilationResult** result, in Compile()
184 XlaCompiler::Options options; in Compile()
190 XlaCompiler::CompileOptions compile_options; in Compile()
202 std::vector<XlaCompiler::Argument> args; in Compile()
212 const XlaCompiler::CompilationResult* result; in Compute()
Dxla_compile_on_demand_op.h40 XlaCompiler::Argument CreateCompilerArgument(OpKernelContext* ctx, int64 i);
46 const XlaCompiler::CompilationResult** result,
49 const XlaCompiler::CompilationResult* result,
Dxla_device.h63 XlaCompiler::ShapeRepresentationFn shape_representation_fn,
72 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn() const { in shape_representation_fn()
83 XlaCompiler::ShapeRepresentationFn shape_representation_fn_;
123 XlaCompiler::ShapeRepresentationFn shape_representation_fn;
226 const XlaCompiler::ShapeRepresentationFn shape_representation_fn_;
Dxla_launch_util.cc208 OpKernelContext* ctx, const XlaCompiler::CompilationResult* kernel, in PopulateInputs()
261 OpKernelContext* ctx, const XlaCompiler::CompilationResult* kernel, in PopulateOutputs()
390 const XlaCompiler::ResourceUpdate& write = kernel->resource_updates[i]; in PopulateOutputs()
412 const XlaCompiler::ResourceUpdate& write = kernel->resource_updates[i]; in PopulateOutputs()
446 std::vector<XlaCompiler::Argument>* args) { in BuildXlaCompilerArguments()
450 XlaCompiler::Argument& arg = (*args)[input_num]; in BuildXlaCompilerArguments()
455 arg.kind = XlaCompiler::Argument::kConstant; in BuildXlaCompilerArguments()
464 arg.kind = XlaCompiler::Argument::kParameter; in BuildXlaCompilerArguments()
466 arg.kind = XlaCompiler::Argument::kConstant; in BuildXlaCompilerArguments()
477 arg.kind = XlaCompiler::Argument::kResource; in BuildXlaCompilerArguments()
Dxla_device_context.h60 XlaCompiler::ShapeRepresentationFn shape_representation_fn,
82 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn() const { in shape_representation_fn()
111 XlaCompiler::ShapeRepresentationFn shape_representation_fn_;
Dxla_launch_util.h154 std::vector<XlaCompiler::Argument>* args);
164 const XlaCompiler::CompilationResult* kernel,
179 const XlaCompiler::CompilationResult* kernel,
Dxla_compilation_cache_test.cc26 std::vector<XlaCompiler::Argument> args(1); in TEST()
27 args[0].kind = XlaCompiler::Argument::kConstant; in TEST()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dif_op.cc58 std::vector<XlaCompiler::Argument> arguments(input_types_.size()); in Compile()
61 XlaCompiler::Argument& arg = arguments[i]; in Compile()
69 arg.kind = XlaCompiler::Argument::kResource; in Compile()
88 arg.kind = XlaCompiler::Argument::kParameter; in Compile()
97 XlaCompiler::CompileOptions options; in Compile()
103 XlaCompiler* compiler = ctx->compiler(); in Compile()
105 XlaCompiler::CompilationResult then_result; in Compile()
108 XlaCompiler::CompilationResult else_result; in Compile()
113 for (XlaCompiler::CompilationResult* result : {&then_result, &else_result}) { in Compile()
114 for (const XlaCompiler::ResourceUpdate& update : result->resource_updates) { in Compile()
[all …]
Dwhile_op.cc43 XlaOpKernelContext* ctx, std::vector<XlaCompiler::Argument>* args, in MakeXlaCompilerArgumentsFromInputs()
54 XlaCompiler::Argument& arg = (*args)[i]; in MakeXlaCompilerArgumentsFromInputs()
63 arg.kind = XlaCompiler::Argument::kResource; in MakeXlaCompilerArgumentsFromInputs()
85 arg.kind = XlaCompiler::Argument::kParameter; in MakeXlaCompilerArgumentsFromInputs()
124 std::vector<XlaCompiler::Argument>* args, in ConvertLoopInvariantsToConst()
130 XlaCompiler::Argument& arg = (*args)[i]; in ConvertLoopInvariantsToConst()
134 if (arg.kind != XlaCompiler::Argument::kResource && loop_invariants[i]) { in ConvertLoopInvariantsToConst()
140 arg.kind = XlaCompiler::Argument::kConstant; in ConvertLoopInvariantsToConst()
155 const XlaCompiler::CompilationResult& body, bool has_token_input_output) { in VerifyBodyInputAndOutputShapeMatch()
180 XlaOpKernelContext* ctx, const XlaCompiler::CompilationResult& cond) { in BuildWrappedCond()
[all …]
Dcase_op.cc55 std::vector<XlaCompiler::Argument> arguments(input_types_.size()); in Compile()
58 XlaCompiler::Argument& arg = arguments[i]; in Compile()
66 arg.kind = XlaCompiler::Argument::kResource; in Compile()
85 arg.kind = XlaCompiler::Argument::kParameter; in Compile()
94 XlaCompiler::CompileOptions options; in Compile()
100 XlaCompiler* compiler = ctx->compiler(); in Compile()
102 std::vector<XlaCompiler::CompilationResult> branch_results(num_branches); in Compile()
103 std::vector<XlaCompiler::CompilationResult*> branch_results_p(num_branches); in Compile()
112 for (XlaCompiler::CompilationResult* result : branch_results_p) { in Compile()
113 for (const XlaCompiler::ResourceUpdate& update : result->resource_updates) { in Compile()
[all …]
Dxla_select_and_scatter_op.cc61 XlaCompiler::CompileOptions compile_options; in Compile()
68 XlaCompiler::Argument select_arg; in Compile()
69 select_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
73 XlaCompiler::CompilationResult select; in Compile()
89 XlaCompiler::Argument scatter_arg; in Compile()
90 scatter_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
94 XlaCompiler::CompilationResult scatter; in Compile()
Dxla_reduce_op.cc60 XlaCompiler::Argument reducer_arg; in Compile()
61 reducer_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
65 XlaCompiler::CompileOptions compile_options; in Compile()
70 XlaCompiler::CompilationResult reducer; in Compile()
Dreduce_window_op.cc77 XlaCompiler::Argument reducer_arg; in Compile()
78 reducer_arg.kind = XlaCompiler::Argument::kParameter; in Compile()
82 XlaCompiler::CompileOptions compile_options; in Compile()
87 XlaCompiler::CompilationResult reducer; in Compile()
Dsendrecv_ops.cc45 XlaCompiler* compiler = ctx->compiler(); in Compile()
76 XlaCompiler* compiler = ctx->compiler(); in Compile()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc115 const XlaCompiler::CompilationResult* compilation_result, in XlaExecutableClosure()
129 const XlaCompiler::CompilationResult* compilation_result() const { in compilation_result()
140 const XlaCompiler::CompilationResult* compilation_result_;
259 const XlaCompiler::CompilationResult** kernel, in CompileToLocalExecutable()
282 XlaCompiler::Options options; in CompileToLocalExecutable()
303 XlaCompiler::CompileOptions compile_options; in CompileToLocalExecutable()
314 std::vector<XlaCompiler::Argument> args; in CompileToLocalExecutable()
328 const XlaCompiler::CompilationResult* kernel; in Compute()
445 const XlaCompiler::CompilationResult* kernel; in Compute()

12