Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc56 XlaCompiler::Options DefaultOptions() { in DefaultOptions()
57 XlaCompiler::Options options; in DefaultOptions()
64 FunctionLibraryDefinition* LocalFlibDef(XlaCompiler* compiler) { in LocalFlibDef()
169 XlaCompiler compiler(DefaultOptions()); in TEST_F()
172 XlaCompiler::CompilationResult result; in TEST_F()
173 TF_ASSERT_OK(compiler.CompileGraph(XlaCompiler::CompileOptions(), "add", in TEST_F()
192 std::vector<XlaCompiler::Argument> args(2); in TEST_F()
193 args[0].kind = XlaCompiler::Argument::kParameter; in TEST_F()
196 args[1].kind = XlaCompiler::Argument::kParameter; in TEST_F()
201 XlaCompiler compiler(DefaultOptions()); in TEST_F()
[all …]
Dxla_compiler.cc50 const std::vector<XlaCompiler::Argument>& args) { in CheckSignature()
67 bool XlaCompiler::Argument::operator==( in operator ==()
68 const XlaCompiler::Argument& other) const { in operator ==()
85 XlaCompiler::XlaCompiler(XlaCompiler::Options options) in XlaCompiler() function in tensorflow::XlaCompiler
120 XlaCompiler::~XlaCompiler() = default;
122 int64 XlaCompiler::NextStepId() { return next_step_id_++; } in NextStepId()
124 uint64 XlaCompiler::SignatureHash::operator()( in operator ()()
141 Status XlaCompiler::FindFunctionBody(const NameAttrList& function, in FindFunctionBody()
158 std::unique_ptr<Graph> XlaCompiler::GetGraph(const FunctionBody* fbody) { in GetGraph()
173 Status XlaCompiler::CompileFunction(const XlaCompiler::CompileOptions& options, in CompileFunction()
[all …]
Dgraph_compiler.cc52 std::vector<XlaCompiler::Argument>* args) { in PrepareArguments()
61 XlaCompiler::Argument& arg = (*args)[i]; in PrepareArguments()
69 arg.kind = XlaCompiler::Argument::kConstant; in PrepareArguments()
72 arg.kind = XlaCompiler::Argument::kConstant; in PrepareArguments()
80 arg.kind = XlaCompiler::Argument::kParameter; in PrepareArguments()
180 XlaCompiler* compiler = xla_op_context.compiler(); in CompileFunctionalNode()
199 std::vector<XlaCompiler::Argument> arguments; in CompileFunctionalNode()
208 XlaCompiler::CompilationResult result; in CompileFunctionalNode()
210 TF_RETURN_IF_ERROR(compiler->CompileFunction(XlaCompiler::CompileOptions(), in CompileFunctionalNode()
217 if (arguments[i].kind == XlaCompiler::Argument::kConstant) { in CompileFunctionalNode()
Dxla_compiler.h84 class XlaCompiler {
271 explicit XlaCompiler(Options options);
273 ~XlaCompiler();
314 const std::vector<XlaCompiler::Argument>& args,
362 TF_DISALLOW_COPY_AND_ASSIGN(XlaCompiler);
Dtf2xla.cc237 std::vector<XlaCompiler::Argument>* xla_args) { in CreateXlaArgs()
241 XlaCompiler::Argument arg; in CreateXlaArgs()
242 arg.kind = XlaCompiler::Argument::kParameter; in CreateXlaArgs()
260 std::vector<XlaCompiler::Argument> xla_args; in ConvertGraphToXla()
264 XlaCompiler::Options compiler_options; in ConvertGraphToXla()
271 XlaCompiler compiler(compiler_options); in ConvertGraphToXla()
273 XlaCompiler::CompilationResult result; in ConvertGraphToXla()
274 TF_RETURN_IF_ERROR(compiler.CompileGraph(XlaCompiler::CompileOptions(), in ConvertGraphToXla()
Dxla_context.h46 XlaContext(XlaCompiler* compiler, xla::ComputationBuilder* builder,
54 XlaCompiler* compiler() const { return compiler_; } in compiler()
120 XlaCompiler* const compiler_;
Dxla_op_kernel.h201 XlaCompiler* compiler() const;
Dxla_context.cc66 XlaCompiler* compiler, xla::ComputationBuilder* builder, in XlaContext()
Dxla_op_kernel.cc441 XlaCompiler* XlaOpKernelContext::compiler() const { in compiler()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.cc138 std::vector<XlaCompiler::Argument>* args) { in BuildArguments()
148 XlaCompiler::Argument& arg = (*args)[input_num]; in BuildArguments()
149 arg.kind = XlaCompiler::Argument::kConstant; in BuildArguments()
164 XlaCompiler::Argument& arg = (*args)[input_num]; in BuildArguments()
166 arg.kind = XlaCompiler::Argument::kParameter; in BuildArguments()
168 arg.kind = XlaCompiler::Argument::kConstant; in BuildArguments()
182 XlaCompiler::Argument& arg = (*args)[input_num]; in BuildArguments()
185 arg.kind = XlaCompiler::Argument::kResource; in BuildArguments()
210 const XlaCompiler::Options& options, in BuildExecutable()
211 const XlaCompiler::CompilationResult& result, in BuildExecutable()
[all …]
Dxla_compilation_cache.h64 Status Compile(const XlaCompiler::Options& options,
68 const XlaCompiler::CompilationResult** compilation_result,
70 const XlaCompiler::CompileOptions* compile_options);
80 Status BuildExecutable(const XlaCompiler::Options& options,
81 const XlaCompiler::CompilationResult& result,
122 XlaCompiler::CompilationResult compilation_result GUARDED_BY(mu);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc35 XlaOpKernelContext* ctx, std::vector<XlaCompiler::Argument>* args, in MakeXlaCompilerArgumentsFromInputs()
45 XlaCompiler::Argument& arg = (*args)[i]; in MakeXlaCompilerArgumentsFromInputs()
54 arg.kind = XlaCompiler::Argument::kResource; in MakeXlaCompilerArgumentsFromInputs()
76 arg.kind = XlaCompiler::Argument::kParameter; in MakeXlaCompilerArgumentsFromInputs()
97 std::vector<XlaCompiler::Argument> arguments; in Compile()
105 XlaCompiler* compiler = ctx->compiler(); in Compile()
117 XlaCompiler::CompileOptions body_options; in Compile()
122 XlaCompiler::CompilationResult body; in Compile()
148 const XlaCompiler::ResourceUpdate& update = body.resource_updates[i]; in Compile()
152 XlaCompiler::Argument& arg = arguments[update.input_index]; in Compile()
[all …]
Dsendrecv_ops.cc45 XlaCompiler* compiler = XlaContext::Get(ctx).compiler(); in Compile()
76 XlaCompiler* compiler = XlaContext::Get(ctx).compiler(); in Compile()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_launch_op.cc255 XlaCompiler::Options options; in Compute()
263 const XlaCompiler::CompilationResult* kernel; in Compute()
375 const XlaCompiler::ResourceUpdate& write = kernel->resource_updates[i]; in Compute()