Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
H A 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 …]
H A 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
H A 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
H A Dgpu_input_fusible_slice_test.cc66 auto hlo_module = in TEST_F() local
113 auto hlo_module = in TEST_F() local
H A Dgpu_spmd_e2e_compile_test.cc44 auto hlo_module = in TEST_F() local
71 auto hlo_module = in TEST_F() local
H A Dgpu_ftz_test.cc45 auto hlo_module = CreateNewVerifiedModuleWithFTZ(ftz_); in CreateBinaryOpModule() local
60 auto hlo_module = CreateNewVerifiedModuleWithFTZ(ftz_); in CreateUnaryOpModule() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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 …]
H A 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
H A Dexecutable.h233 explicit Executable(std::shared_ptr<HloModule> hlo_module) in Executable()
238 std::shared_ptr<HloModule> hlo_module, in Executable()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A 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 …]
H A 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 …]
H A 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
H A 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()
H A Dcodegen_test_base.cc21 std::unique_ptr<HloModule> hlo_module) { in CompileToExecutable()
33 std::unique_ptr<HloModule> hlo_module, in CompileToAotCompilationResult()
H A Dmultioutput_fusion_test.cc63 auto hlo_module = CreateNewVerifiedModule(); in RunTest2D() local
123 auto hlo_module = CreateNewVerifiedModule(); in RunTest1D() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_compiler.cc337 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloModule()
693 Status GpuCompiler::PrepareHloModuleForIrEmitting(HloModule* hlo_module) { in PrepareHloModuleForIrEmitting()
723 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
854 const HloModule* hlo_module) { in AssignBuffers()
880 llvm::ArrayRef<int64_t> buffer_sizes, HloModule* hlo_module, in LowerToJitRt()
966 HloModule* hlo_module, llvm::LLVMContext* llvm_context, in CompileModuleToLlvmIrImpl()
1520 HloModule* hlo_module, llvm::LLVMContext* llvm_context, in CompileModuleToLlvmIr()
H A Damdgpu_compiler.cc77 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
109 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
H A Dir_emitter_context.h39 IrEmitterContext(const HloModule* hlo_module, in IrEmitterContext()
58 const HloModule& hlo_module() const { return *hlo_module_; } in hlo_module() function
H A Dnvptx_compiler.cc72 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloConvolutionCanonicalization()
120 HloModule* hlo_module, se::StreamExecutor* stream_exec, in OptimizeHloPostLayoutAssignment()
H A Dgpu_executable.cc966 GetOutputInfo(const HloModule& hlo_module, const BufferAssignment& assignment) { in GetOutputInfo()
1007 std::shared_ptr<HloModule> hlo_module, GpuVersion gpu_version, in GpuExecutable()
1026 std::shared_ptr<HloModule> hlo_module, absl::string_view obj_file, in LoadFromObjFile()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dxla_mlir_translate_registration.cc134 auto hlo_module = computation.proto(); in ConvertMlirHloToHloViaBuilder() local
166 HloModule* hlo_module = statusOrHloModule.ValueOrDie().get(); in MlirHloToHloTextTranslateFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_on_demand_compiler.cc58 XLA_HloModule hlo_module; in RunHloPasses() local
86 XLA_HloModule hlo_module; in RunBackend() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dtf_xla_mlir_translate.cc94 xla::HloModule* hlo_module = status_or_hlo_module.ValueOrDie().get(); in PrintHloModuleText() local
292 auto hlo_module = computation.proto(); in CompileMlirToXlaHloViaBuilder() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_compilation_cache.cc521 const xla::HloModuleProto& hlo_module = in CompileStrict() local
816 const XlaSerializedCacheKey& key, const xla::HloModuleProto& hlo_module, in VerifyLoadedCacheEntry()
864 const xla::HloModuleProto& hlo_module = in SerializeEntry() local

1234