Searched refs:flib_runtime_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.h | 285 FunctionLibraryRuntime* flib_runtime() const { return flib_runtime_; } in flib_runtime()
340 FunctionLibraryRuntime* flib_runtime_; // owned by pflr_. variable
|
| D | xla_compiler.cc | 523 flib_runtime_ = pflr_->GetFLR(device_->name()); in XlaCompiler()
564 GetFunctionBody(function, flib_runtime_, fbody), in FindFunctionBody()
567 *config_proto = flib_runtime_->config_proto(); in FindFunctionBody()
643 flib_runtime_->GetFunctionLibraryDefinition(), &shape_info) in GetGraph()
659 optimizer.Optimize(flib_runtime_, flib_runtime_->env(), in GetGraph()
666 flib_runtime_->GetFunctionLibraryDefinition(), &shape_info) in GetGraph()
682 optimizer.Optimize(flib_runtime_, flib_runtime_->env(), in GetGraph()
1300 flib_runtime_->GetFunctionLibraryDefinition()); in CompileGraph()
1365 flib_runtime_, NextStepId())); in CompileGraph()
|