Home
last modified time | relevance | path

Searched defs:flib_def (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test_helper.cc23 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in MarkForCompilation()
58 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation() local
Dresource_operation_safety_analysis.cc97 Status MayCallFunction(const Node& n, const FunctionLibraryDefinition* flib_def, in MayCallFunction()
116 const Node& n, const FunctionLibraryDefinition* flib_def, in XlaResourceOpKindForNode()
267 const Graph& g, const FunctionLibraryDefinition* flib_def, in ComputeIncompatibleResourceOperationPairs()
Dresource_operation_safety_analysis_test.cc243 FunctionDefLibrary flib_def; in CreateFunctionDefLibWithConstFunction() local
264 FunctionDefLibrary flib_def = in TEST() local
286 FunctionDefLibrary flib_def = in TEST() local
306 FunctionDefLibrary flib_def = in TEST() local
326 FunctionDefLibrary flib_def = in TEST() local
348 FunctionDefLibrary flib_def = in TEST() local
375 FunctionDefLibrary flib_def = in TEST() local
Dencapsulate_xla_computations_pass_test.cc36 const FunctionLibraryDefinition& flib_def, const string& function) { in MakeOuterGraph()
147 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
206 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
305 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST() local
Dbuild_xla_ops_pass.cc234 const FunctionLibraryDefinition& flib_def, Node* n, Graph* g, in ReplaceFunctionCallWithPartionedCall()
279 const FunctionLibraryDefinition& flib_def, in InferDeviceForCluster()
312 const FunctionLibraryDefinition& flib_def, bool lazy_compilation_enabled, in ReplaceNodeWithXlaCompileAndXlaRun()
Dpartially_decluster_pass.cc282 const FunctionLibraryDefinition* flib_def, in PartiallyDeclusterGraph()
Dxla_cluster_util.cc230 const Graph* graph, const FunctionLibraryDefinition* flib_def, in AdjustCycleDetectionGraphForResourceOps()
Dencapsulate_xla_computations_pass.cc211 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def) { in Encapsulate()
Dbuild_xla_ops_pass_test.cc61 FunctionLibraryDefinition flib_def(graph->op_registry(), fdef_lib); in BuildXlaOps() local
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.h74 FunctionNameGenerator(const FunctionLibraryDefinition* flib_def, in FunctionNameGenerator()
Dgraph_execution_state.h64 std::unique_ptr<FunctionLibraryDefinition> flib_def; member
Doptimization_registry.h43 FunctionLibraryDefinition* flib_def = nullptr; // Not owned. member
Dlower_if_while_test.cc47 FunctionLibraryDefinition flib_def((*graph)->flib_def()); in Rewrite() local
Dlower_while_op_test.cc39 FunctionLibraryDefinition flib_def((*graph)->flib_def()); in Rewrite() local
Dlower_if_op_test.cc47 FunctionLibraryDefinition flib_def((*graph)->flib_def()); in Rewrite() local
Ddirect_session.h172 std::unique_ptr<FunctionLibraryDefinition> flib_def; member
/external/tensorflow/tensorflow/tools/graph_transforms/
Dadd_default_attributes.cc28 std::unique_ptr<FunctionLibraryDefinition> flib_def( in AddDefaultAttributes() local
/external/tensorflow/tensorflow/core/common_runtime/data/
Dstandalone.cc55 std::unique_ptr<FunctionLibraryDefinition> flib_def = in FromGraph() local
113 FunctionLibraryDefinition* flib_def, thread::ThreadPool* pool) in Dataset()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc60 std::unique_ptr<FunctionLibraryDefinition> flib_def, in IteratorResource()
148 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in Restore() local
198 Status AddLibrary(const FunctionLibraryDefinition& flib_def) { in AddLibrary()
226 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in SetIteratorFromDataset() local
287 std::shared_ptr<FunctionLibraryDefinition> flib_def; member
423 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in Compute() local
478 std::unique_ptr<FunctionLibraryDefinition>* flib_def, in CreatePrivateFLR()
511 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in Compute() local
841 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in TryInit() local
Dmulti_device_iterator_ops.cc49 std::unique_ptr<FunctionLibraryDefinition> flib_def, in MultiDeviceIterator()
406 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in Compute() local
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.h51 const FunctionLibraryDefinition* flib_def = nullptr; member
/external/tensorflow/tensorflow/core/util/
Ddump_graph.cc119 const FunctionLibraryDefinition* flib_def, in DumpGraphToFile()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h200 Status EnsureFunctionIsStateless(const FunctionLibraryDefinition& flib_def, in EnsureFunctionIsStateless()
414 const FunctionLibraryDefinition* flib_def = nullptr; // Not owned. member
421 const FunctionLibraryDefinition& flib_def() { return *params_.flib_def; } in flib_def() function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla.cc344 FunctionLibraryDefinition flib_def(OpRegistry::Global(), graph_def.library()); in InitGraph() local
Dxla_compiler.h296 const FunctionLibraryDefinition* flib_def = nullptr; member

12