| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compiler.h | 163 struct CompileOptions { struct
184 const CompileOptions& options) = 0; argument
189 CompileOptions{device_allocator}); in RunHloPasses()
199 const CompileOptions& options) { in RunHloPassesAndBufferAssignement()
213 const CompileOptions& options) = 0;
218 CompileOptions{device_allocator}); in RunBackend()
230 const CompileOptions& options) = 0;
236 CompileOptions{device_allocator}); in Compile()
|
| D | service.h | 238 se::StreamExecutor* executor, const Compiler::CompileOptions& options,
247 const Compiler::CompileOptions& options, bool run_backend_only = false);
|
| D | llvm_compiler.h | 76 const CompileOptions& options) override;
|
| D | llvm_compiler.cc | 28 const CompileOptions& options) { in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_compiler.cc | 438 py::class_<CompileOptions> compile_options(m, "CompileOptions"); in BuildXlaCompilerSubmodule()
440 .def(py::init([]() -> CompileOptions { in BuildXlaCompilerSubmodule()
441 CompileOptions options; in BuildXlaCompilerSubmodule()
449 .def_readwrite("argument_layouts", &CompileOptions::argument_layouts) in BuildXlaCompilerSubmodule()
451 &CompileOptions::parameter_is_tupled_arguments) in BuildXlaCompilerSubmodule()
453 &CompileOptions::executable_build_options) in BuildXlaCompilerSubmodule()
457 &CompileOptions::parameter_is_tupled_arguments) in BuildXlaCompilerSubmodule()
460 [](const CompileOptions& options) { in BuildXlaCompilerSubmodule()
463 [](CompileOptions& options, int num_replicas) { in BuildXlaCompilerSubmodule()
468 [](const CompileOptions& options) { in BuildXlaCompilerSubmodule()
[all …]
|
| D | py_client.h | 136 const XlaComputation& computation, CompileOptions options);
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.h | 48 const CompileOptions& options) override;
51 const CompileOptions& options) override;
55 const CompileOptions& options) override;
|
| D | compiler.cc | 98 const CompileOptions& /*options*/) { in RunHloPasses() argument
106 const CompileOptions& /*options*/) { in RunBackend() argument
131 const CompileOptions& options) { in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.h | 58 const CompileOptions& options) override;
64 const CompileOptions& options) override;
98 const CompileOptions& options) override;
107 se::StreamExecutor* stream_exec, const CompileOptions& options,
170 const Compiler::CompileOptions& options,
|
| D | gpu_compiler.cc | 491 const CompileOptions& options) { in RunHloPasses()
515 bool optimize, const CompileOptions& options) { in RunHloPassesAndBufferAssignement()
679 const CompileOptions& options, in CompileToTargetBinary()
857 const CompileOptions& options) { in RunBackend()
1099 const Compiler::CompileOptions& options, in CompileLmhloToExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.h | 131 const CompileOptions& options) override;
135 const CompileOptions& options) override;
141 const CompileOptions& options) override;
145 const CompileOptions& options) override;
|
| /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()
265 XlaCompiler::CompileOptions compile_options; in TEST_F()
318 XlaCompiler::CompileOptions compile_options; in TEST_F()
364 XlaCompiler::CompileOptions compile_options; 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()
525 XlaCompiler::CompileOptions compile_options; in TEST_F()
560 compiler.CompileGraph(XlaCompiler::CompileOptions(), "reshape", in TEST_F()
[all …]
|
| D | xla_compiler.h | 106 struct CompileOptions { struct
213 Status CompileFunction(const CompileOptions& options,
222 const CompileOptions& options, string const& name,
|
| D | tf2xla.cc | 83 XlaCompiler::CompileOptions options; in ConvertGraphToXla()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache.h | 75 const XlaCompiler::CompileOptions& compile_options,
87 const XlaCompiler::CompileOptions& compile_options,
|
| D | get_compiler_ir.cc | 38 const XlaCompiler::CompileOptions& compile_options, in GetLocalExecutable()
110 XlaCompiler::CompileOptions compile_options; in GetCompilerIr()
|
| D | xla_compilation_cache_test.cc | 73 XlaCompiler::CompileOptions{}, in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_client.h | 115 struct CompileOptions { struct
186 const XlaComputation& computation, CompileOptions options) = 0;
|
| D | gpu_multistream_test.cc | 48 CompileOptions compile_options; in TEST()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 221 const CompileOptions& options) override { in RunHloPasses()
249 const CompileOptions& options) override { in RunBackend()
278 const CompileOptions& options) override { in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_scatter_ops.cc | 77 XlaCompiler::CompileOptions compile_options; in Compile()
|
| D | reduce_window_op.cc | 82 XlaCompiler::CompileOptions compile_options; in Compile()
|
| D | sort_ops.cc | 92 XlaCompiler::CompileOptions compile_options; in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client_extension.cc | 132 const XlaComputation& computation, CompileOptions options) in PYBIND11_MODULE()
141 py::arg("compile_options") = CompileOptions()); in PYBIND11_MODULE()
|
| /external/vulkan-validation-layers/tests/ |
| D | vktestframeworkandroid.cpp | 86 shaderc::CompileOptions options; in GLSLtoSPV()
|