Searched refs:CompileGraph (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 186 TF_ASSERT_OK(compiler.CompileGraph(XlaCompiler::CompileOptions(), "add", in TEST_F()
217 TF_ASSERT_OK(compiler.CompileGraph(XlaCompiler::CompileOptions(), "add", in TEST_F()
268 TF_ASSERT_OK(compiler.CompileGraph(compile_options, "add", std::move(graph), in TEST_F()
320 TF_ASSERT_OK(compiler.CompileGraph(compile_options, "add", std::move(graph), in TEST_F()
366 TF_ASSERT_OK(compiler.CompileGraph(compile_options, "add", std::move(graph), in TEST_F()
414 TF_ASSERT_OK(compiler.CompileGraph(XlaCompiler::CompileOptions(), "add", in TEST_F()
457 TF_ASSERT_OK(compiler.CompileGraph(XlaCompiler::CompileOptions(), "transpose", in TEST_F()
479 TF_ASSERT_OK(compiler.CompileGraph(XlaCompiler::CompileOptions(), "compile", in TEST_F()
528 TF_ASSERT_OK(compiler.CompileGraph(compile_options, "add", std::move(graph), in TEST_F()
560 compiler.CompileGraph(XlaCompiler::CompileOptions(), "reshape", in TEST_F()
[all …]
|
| D | tf2xla.cc | 85 TF_RETURN_IF_ERROR(compiler.CompileGraph( in ConvertGraphToXla()
|
| D | xla_compiler.h | 221 Status CompileGraph(
|
| D | xla_compiler.cc | 824 CompileGraph(options, function_id, std::move(graph), args, result)); in CompileFunction()
1282 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 | 107 Status CompileGraph(GraphDef graph_def, const tf2xla::Config& config, in CompileGraph() function
237 CompileGraph(std::move(graph_def), config, flags, &compile_result); in Main()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.cc | 295 return compiler->CompileGraph(compile_options, node_def.name(), in CompileSingleOp()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 385 return compiler->CompileGraph(compile_options, function_id, std::move(graph), in CompileTFFunctionToHlo()
|