Searched refs:CompileGraph (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_compiler_test.cc | 186 TF_ASSERT_OK(compiler.CompileGraph( in TEST_F() 217 TF_ASSERT_OK(compiler.CompileGraph(XlaCompiler::CompileOptions(), "add", in TEST_F() 269 TF_ASSERT_OK(compiler.CompileGraph(compile_options, "add", std::move(graph), in TEST_F() 321 TF_ASSERT_OK(compiler.CompileGraph(compile_options, "add", std::move(graph), in TEST_F() 368 TF_ASSERT_OK(compiler.CompileGraph(compile_options, "add", std::move(graph), in TEST_F() 416 TF_ASSERT_OK(compiler.CompileGraph(XlaCompiler::CompileOptions(), "add", in TEST_F() 458 TF_ASSERT_OK(compiler.CompileGraph(XlaCompiler::CompileOptions(), "transpose", in TEST_F() 509 TF_ASSERT_OK(compiler.CompileGraph(compile_options, "add", std::move(graph), in TEST_F() 540 Status status = compiler.CompileGraph(XlaCompiler::CompileOptions(), in TEST_F() 583 TF_ASSERT_OK(compiler.CompileGraph(compile_options, "constants", in TEST_F() [all …]
|
D | tf2xla.cc | 81 TF_RETURN_IF_ERROR(compiler.CompileGraph(XlaCompiler::CompileOptions(), in ConvertGraphToXla()
|
D | xla_compiler.h | 367 Status CompileGraph(
|
D | xla_compiler.cc | 753 CompileGraph(options, function_id, std::move(graph), args, {}, result)); in CompileFunction() 1195 return CompileGraph(options, node_def.name(), std::move(graph), args, {}, in CompileSingleOp() 1291 Status XlaCompiler::CompileGraph( in CompileGraph() function in tensorflow::XlaCompiler
|
/external/tensorflow/tensorflow/compiler/aot/ |
D | compile.h | 45 Status CompileGraph(GraphDef graph_def, const tf2xla::Config& config,
|
D | compile.cc | 96 Status CompileGraph(GraphDef graph_def, const tf2xla::Config& config, in CompileGraph() function 197 CompileGraph(std::move(graph_def), config, flags, &compile_result)); in Main()
|