Home
last modified time | relevance | path

Searched refs:flib (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.h59 Graph* graph, FunctionLibraryRuntime* flib, in GraphCompiler() argument
64 flib_(flib), in GraphCompiler()
Dxla_compiler_test.cc52 FunctionDefLibrary flib; in SetUp() local
53 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib)); in SetUp()
625 FunctionDefLibrary flib; in TEST_F() local
626 *flib.add_function() = FillFn(); in TEST_F()
635 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib)); in TEST_F()
Dxla_compiler.cc296 XlaCompilationDevice* device, FunctionLibraryRuntime* flib, in ExecuteGraph() argument
315 GraphCompiler graph_compiler(xla_context, device, graph.get(), flib, in ExecuteGraph()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc54 FunctionDefLibrary flib; in MarkForCompilation() local
55 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation()
200 FunctionDefLibrary flib; in TEST() local
201 *flib.add_function() = compilable; in TEST()
202 *flib.add_function() = uncompilable; in TEST()
203 *flib.add_function() = noinline; in TEST()
204 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.h52 explicit ClientGraph(std::unique_ptr<FunctionLibraryDefinition> flib, in ClientGraph()
54 : flib_def(std::move(flib)), in ClientGraph()
Dgraph_execution_state.cc418 std::unique_ptr<FunctionLibraryDefinition> flib( in BuildGraph() local
425 optimization_options.flib_def = flib.get(); in BuildGraph()
435 new ClientGraph(std::move(flib), rewrite_metadata.feed_types, in BuildGraph()
Dprocess_function_library_runtime_test.cc34 void Init(const std::vector<FunctionDef>& flib) { in Init() argument
42 for (const auto& fdef : flib) *(proto.add_function()) = fdef; in Init()
Dexecutor.h231 Status CreateNonCachedKernel(Device* device, FunctionLibraryRuntime* flib,
Dconstant_folding_test.cc324 FunctionDefLibrary flib; in TEST_F() local
325 *flib.add_function() = test::function::XTimesTwo(); in TEST_F()
327 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST_F()
332 TF_EXPECT_OK(s.graph()->AddFunctionLibrary(flib)); in TEST_F()
Ddirect_session.h116 FunctionLibraryRuntime* flib = nullptr; // not owned. member
Ddirect_session_test.cc926 FunctionLibraryDefinition flib(OpRegistry::Global(), library_graph_def); in TestSessionInterOpThreadsImpl() local
927 Graph g(&flib); in TestSessionInterOpThreadsImpl()
Ddirect_session.cc685 const string device = partition.flib->device()->name(); in Run()
1200 item->flib = lib; in GetOrCreateExecutors()
Dfunction_test.cc138 void Init(const std::vector<FunctionDef>& flib) { in Init() argument
146 for (const auto& fdef : flib) *(proto.add_function()) = fdef; in Init()
Dexecutor.cc2621 Status CreateNonCachedKernel(Device* device, FunctionLibraryRuntime* flib, in CreateNonCachedKernel() argument
2626 return CreateOpKernel(device_type, device, allocator, flib, ndef, in CreateNonCachedKernel()
/external/tensorflow/tensorflow/c/eager/
Druntime.cc253 Status KernelAndDevice::Init(const NodeDef& ndef, FunctionLibraryRuntime* flib, in Init() argument
256 Status s = flib->CreateKernel(ndef, &k); in Init()
257 out->device_ = flib->device(); in Init()
259 out->flib_ = flib; in Init()
Druntime.h168 static Status Init(const NodeDef& ndef, FunctionLibraryRuntime* flib,
/external/tensorflow/tensorflow/contrib/tensorrt/segment/
Dsegment.cc124 tensorflow::FunctionLibraryDefinition flib(tensorflow::OpRegistry::Global(), in SegmentGraph() local
126 tensorflow::Graph graph(flib); in SegmentGraph()
/external/tensorflow/tensorflow/contrib/tensorrt/convert/
Dconvert_graph.cc231 tensorflow::FunctionLibraryDefinition flib(tensorflow::OpRegistry::Global(), in ConvertGraphDefToTensorRT() local
233 tensorflow::Graph graph(flib); in ConvertGraphDefToTensorRT()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc185 const NodeDef* node_def, const OpDef* op_def, FunctionLibraryRuntime* flib, in OpKernelConstruction() argument
195 flib_(flib), in OpKernelConstruction()
1109 Allocator* allocator, FunctionLibraryRuntime* flib, in CreateOpKernel() argument
1166 device_type, device, allocator, &node_def, op_def, flib, inputs, in CreateOpKernel()
Dop_kernel.h236 const OpDef* op_def, FunctionLibraryRuntime* flib,
1166 Allocator* allocator, FunctionLibraryRuntime* flib,
/external/syslinux/com32/lua/src/
Dliolib.c627 static const luaL_Reg flib[] = { variable
647 luaL_setfuncs(L, flib, 0); /* add file methods to new metatable */ in createmeta()