Home
last modified time | relevance | path

Searched defs:hlo_module (Results 1 – 25 of 80) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_kernel_tiling_test.cc79 auto hlo_module = in TEST_F() local
107 auto hlo_module = in TEST_F() local
151 auto hlo_module = in TEST_F() local
186 auto hlo_module = in TEST_F() local
222 auto hlo_module = in TEST_F() local
250 auto hlo_module = in TEST_F() local
278 auto hlo_module = in TEST_F() local
314 auto hlo_module = in TEST_F() local
346 auto hlo_module = in TEST_F() local
390 auto hlo_module = in TEST_F() local
[all …]
Dgpu_unrolling_test.cc51 auto hlo_module = in TEST_F() local
72 auto hlo_module = in TEST_F() local
94 auto hlo_module = in TEST_F() local
127 auto hlo_module = in TEST_F() local
158 auto hlo_module = in TEST_F() local
192 auto hlo_module = in TEST_F() local
226 auto hlo_module = in TEST_F() local
251 auto hlo_module = in TEST_F() local
289 auto hlo_module = in TEST_F() local
Dgpu_ldg_test.cc55 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
85 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
112 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_spmd_e2e_compile_test.cc44 auto hlo_module = in TEST_F() local
71 auto hlo_module = in TEST_F() local
Dgpu_input_fusible_slice_test.cc66 auto hlo_module = in TEST_F() local
113 auto hlo_module = in TEST_F() local
Dgpu_ftz_test.cc45 auto hlo_module = CreateNewVerifiedModuleWithFTZ(ftz_); in CreateBinaryOpModule() local
60 auto hlo_module = CreateNewVerifiedModuleWithFTZ(ftz_); in CreateUnaryOpModule() local
Dgpu_dyn_shape_test.cc35 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_spmd_single_partition_test.cc43 auto hlo_module = in TEST_F() local
Dgpu_copy_test.cc47 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
Dgpu_codegen_test.cc49 std::unique_ptr<VerifiedHloModule> hlo_module, absl::string_view pattern) { in CompileAndOptionallyVerifyPtx()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_test.cc44 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
60 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
83 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
104 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
123 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
140 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
160 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
185 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
216 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
232 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
[all …]
Dcpu_gpu_fusion_test.cc89 auto hlo_module = CreateNewVerifiedModule(); in TestElementwise2D() local
199 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
247 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
282 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
306 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
330 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
345 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
360 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
375 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
390 auto hlo_module = CreateNewVerifiedModule(); in XLA_TEST_F() local
[all …]
Dcodegen_test_base.cc21 std::unique_ptr<HloModule> hlo_module) { in CompileToExecutable()
33 std::unique_ptr<HloModule> hlo_module, in CompileToAotCompilationResult()
Dllvm_irgen_test_base.cc50 std::unique_ptr<HloModule> hlo_module, const std::string& pattern, in CompileAndVerifyIr()
73 std::unique_ptr<HloModule> hlo_module, const AotCompilationOptions& options, in CompileAheadOfTimeAndVerifyIr()
110 std::unique_ptr<HloModule> hlo_module) { in GetOptimizedModule()
Dllvm_compiler_test.cc47 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
53 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
115 auto hlo_module = CreateNewVerifiedModule(); in TestCompilerHooks() local
137 std::unique_ptr<HloModule> hlo_module = CreateNewVerifiedModule(); in TestMultiModuleCompilation() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc146 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
180 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
216 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
246 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
273 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
303 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
345 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
380 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
403 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
426 auto hlo_module = BuildHloGraph(&builder); in TEST_F() local
[all …]
Dmap_inliner_test.cc61 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
95 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
133 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
165 auto hlo_module = CreateNewVerifiedModule(); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc82 Status InterpreterCompiler::RunHloOptimization(HloModule* hlo_module) { in RunHloOptimization()
102 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
110 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* stream_exec, in RunBackend()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Damdgpu_compiler.cc77 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
109 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
Dir_emitter_context.h39 IrEmitterContext(const HloModule* hlo_module, in IrEmitterContext()
58 const HloModule& hlo_module() const { return *hlo_module_; } in hlo_module() function
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dtpu_executable_interface.h43 explicit TpuExecutableInterface(std::shared_ptr<HloModule> hlo_module) in TpuExecutableInterface()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compilation_cache.proto40 xla.HloModuleProto hlo_module = 2; field
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dxla_mlir_translate_registration.cc134 auto hlo_module = computation.proto(); in ConvertMlirHloToHloViaBuilder() local
166 HloModule* hlo_module = statusOrHloModule.ValueOrDie().get(); in MlirHloToHloTextTranslateFunction() local
Dhlo_to_mlir_hlo.cc32 Status ConvertHloToMlirHlo(mlir::ModuleOp module, xla::HloModule* hlo_module, in ConvertHloToMlirHlo()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/translate/
Dconvert_xla_gpu.cc34 std::unique_ptr<xla::HloModule> hlo_module, tfrt::HostContext* host, in ConvertXlaGpuToGpuProgram()

1234