Home
last modified time | relevance | path

Searched refs:module_ (Results 1 – 25 of 168) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_dimension_inference_test.cc45 module_ = CreateNewVerifiedModule(); in DynamicDimensionInferenceTest()
56 DynamicDimensionInference::Run(module_.get(), handler, shape_check_mode, in RunInference()
71 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetAdd()
91 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetAddTuple()
102 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetGe()
105 std::unique_ptr<HloModule> module_; member in xla::__anon0e1f7b030111::DynamicDimensionInferenceTest
119 module_->AddEntryComputation(builder.Build()); in TEST_F()
120 SCOPED_TRACE(module_->ToString()); in TEST_F()
123 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
140 module_->AddEntryComputation(builder.Build()); in TEST_F()
[all …]
Dhlo_alias_analysis_test.cc44 module_ = CreateNewVerifiedModule(); in HloAliasAnalysisTest()
50 analysis_ = HloAliasAnalysis::Run(module_.get(), in RunAnalysis()
86 DependencyHloOrdering ordering(module_.get()); in AnyValuesInSameBufferInterfere()
144 std::unique_ptr<HloModule> module_; member in xla::__anonb9c7801a0111::HloAliasAnalysisTest
159 module_->AddEntryComputation(builder.Build()); in TEST_F()
160 SCOPED_TRACE(module_->ToString()); in TEST_F()
194 module_->AddEntryComputation(builder.Build()); in TEST_F()
195 SCOPED_TRACE(module_->ToString()); in TEST_F()
238 module_->AddEntryComputation(builder.Build()); in TEST_F()
239 SCOPED_TRACE(module_->ToString()); in TEST_F()
[all …]
Dhlo_live_range_test.cc39 HloLiveRangeTest() : module_(CreateNewVerifiedModule()) {} in HloLiveRangeTest()
43 alias_analysis_ = HloAliasAnalysis::Run(module_.get()).ValueOrDie(); in Analyze()
45 module_->entry_computation()) in Analyze()
49 std::unique_ptr<HloModule> module_; member in xla::__anond24688290111::HloLiveRangeTest
95 module_->AddEntryComputation(builder.Build()); in TEST_F()
97 HloSchedule schedule(module_.get()); in TEST_F()
99 schedule.set_sequence(module_->entry_computation(), {paramA, paramX, mul}); in TEST_F()
124 module_->AddEntryComputation(builder.Build()); in TEST_F()
126 HloSchedule schedule(module_.get()); in TEST_F()
128 schedule.set_sequence(module_->entry_computation(), in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc53 HloDataflowAnalysisTest() : module_(CreateNewVerifiedModule()) {} in HloDataflowAnalysisTest()
61 EXPECT_TRUE(async_op_canonicalizer.Run(module_.get()).ok()); in RunAnalysis()
64 EXPECT_TRUE(dce.Run(module_.get()).ok()); in RunAnalysis()
67 EXPECT_TRUE(flatten.Run(module_.get()).ok()); in RunAnalysis()
69 HloDataflowAnalysis::Run(*module_, ssa_form, bitcast_defines_value) in RunAnalysis()
102 std::unique_ptr<HloModule> module_; member in xla::__anon676be1220111::HloDataflowAnalysisTest
120 module_->AddEntryComputation(builder.Build()); in TEST_P()
121 SCOPED_TRACE(module_->ToString()); in TEST_P()
169 module_->AddEntryComputation(builder.Build()); in TEST_P()
170 SCOPED_TRACE(module_->ToString()); in TEST_P()
[all …]
Ddynamic_padder_test.cc85 DynamicPadderTest() : HloTestBase() { module_ = CreateNewVerifiedModule(); } in DynamicPadderTest()
99 return RunHloPass(&padder, module_.get()); in RunPadder()
116 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetScalarAddComputation()
119 std::unique_ptr<HloModule> module_; member in xla::__anond6af44bf0111::DynamicPadderTest
173 EXPECT_FALSE(module_->is_dynamic()); in TEST_F()
174 module_->AddEntryComputation(builder.Build()); in TEST_F()
177 TF_CHECK_OK(module_->dynamic_parameter_binding().Bind( in TEST_F()
184 EXPECT_TRUE(module_->is_dynamic()); in TEST_F()
205 module_ = GetHloModule(hlo_text); in TEST_F()
224 module_->entry_computation()->GetInstructionWithName("custom-call.1"); in TEST_F()
[all …]
Dcopy_insertion_test.cc335 WhileCopyInsertionTest() : module_(CreateNewVerifiedModule()) {} in WhileCopyInsertionTest()
587 module_->AddEntryComputation(builder.Build()); in BuildWhileInstruction()
595 module_->AddEntryComputation(builder.Build()); in BuildWhileInstruction()
671 auto condition = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit()
673 auto body = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit()
679 module_->AddEntryComputation(builder->Build()); in BuildWhileInstructionWithCustomInit()
683 std::unique_ptr<HloModule> module_; member in xla::__anon62fae9200111::WhileCopyInsertionTest
705 auto condition = module_->AddEmbeddedComputation( in TEST_F()
708 module_->AddEmbeddedComputation(BuildIndependentBodyComputation()); in TEST_F()
711 InsertCopies(module_.get()); in TEST_F()
[all …]
Dtuple_points_to_analysis_test.cc55 module_ = CreateNewVerifiedModule(); in BuildModule()
56 module_->AddEntryComputation(std::move(computation)); in BuildModule()
60 CHECK_NOTNULL(module_.get()); in RunAnalysis()
61 points_to_analysis_ = TuplePointsToAnalysis::Run(module_.get()).value(); in RunAnalysis()
122 std::unique_ptr<HloModule> module_; member in xla::__anon8b9a5a960111::TuplePointsToAnalysisTest
/external/python/pybind11/tests/test_embed/
Dtest_interpreter.cpp54 auto module_ = py::module_::import("test_interpreter"); variable
55 REQUIRE(py::hasattr(module_, "DerivedWidget"));
57 auto locals = py::dict("hello"_a="Hello, World!", "x"_a=5, **module_.attr("__dict__"));
64 auto py_widget = module_.attr("DerivedWidget")("The question");
73 REQUIRE_NOTHROW(py::module_::import("widget_module"));
74 REQUIRE_THROWS_WITH(py::module_::import("throw_exception"),
76 REQUIRE_THROWS_WITH(py::module_::import("throw_error_already_set"),
110 REQUIRE(py::module_::import("widget_module").attr("add")(1, 2).cast<int>() == 3);
113 REQUIRE(py::module_::import("external_module").attr("A")(123).attr("value").cast<int>() == 123);
117 py::module_::import("external_module").attr("internals_at")().cast<uintptr_t>());
[all …]
/external/deqp-deps/SPIRV-Tools/source/opt/
Dir_loader.cpp36 module_(m), in IrLoader()
150 module_->AddFunction(std::move(function_)); in AddInstruction()
183 module_->AddCapability(std::move(spv_inst)); in AddInstruction()
185 module_->AddExtension(std::move(spv_inst)); in AddInstruction()
187 module_->AddExtInstImport(std::move(spv_inst)); in AddInstruction()
189 module_->SetMemoryModel(std::move(spv_inst)); in AddInstruction()
191 module_->SetSampledImageAddressMode(std::move(spv_inst)); in AddInstruction()
193 module_->AddEntryPoint(std::move(spv_inst)); in AddInstruction()
196 module_->AddExecutionMode(std::move(spv_inst)); in AddInstruction()
198 module_->AddDebug1Inst(std::move(spv_inst)); in AddInstruction()
[all …]
Ddecoration_manager.cpp65 auto context = module_->context(); in RemoveDecorationsFrom()
140 decoration->Clone(module_->context())); in RemoveDecorationsFrom()
142 module_->AddAnnotationInst(std::move(new_inst)); in RemoveDecorationsFrom()
143 auto decoration_iter = --module_->annotation_end(); in RemoveDecorationsFrom()
351 if (!module_) return; in AnalyzeDecorations()
354 for (Instruction& inst : module_->annotations()) { in AnalyzeDecorations()
389 IRContext* ctx = module_->context(); in AddDecoration()
511 auto context = module_->context(); in CloneDecorations()
514 std::unique_ptr<Instruction> new_inst(inst->Clone(module_->context())); in CloneDecorations()
516 module_->AddAnnotationInst(std::move(new_inst)); in CloneDecorations()
[all …]
Dir_loader.h49 Module* module() const { return module_; } in module()
54 module_->SetHeader({magic, version, generator, bound, reserved}); in SetModuleHeader()
74 Module* module_; variable
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dir_loader.cpp37 module_(m), in IrLoader()
152 module_->AddFunction(std::move(function_)); in AddInstruction()
185 module_->AddCapability(std::move(spv_inst)); in AddInstruction()
187 module_->AddExtension(std::move(spv_inst)); in AddInstruction()
189 module_->AddExtInstImport(std::move(spv_inst)); in AddInstruction()
191 module_->SetMemoryModel(std::move(spv_inst)); in AddInstruction()
193 module_->SetSampledImageAddressMode(std::move(spv_inst)); in AddInstruction()
195 module_->AddEntryPoint(std::move(spv_inst)); in AddInstruction()
198 module_->AddExecutionMode(std::move(spv_inst)); in AddInstruction()
200 module_->AddDebug1Inst(std::move(spv_inst)); in AddInstruction()
[all …]
Ddecoration_manager.cpp64 auto context = module_->context(); in RemoveDecorationsFrom()
140 decoration->Clone(module_->context())); in RemoveDecorationsFrom()
142 module_->AddAnnotationInst(std::move(new_inst)); in RemoveDecorationsFrom()
143 auto decoration_iter = --module_->annotation_end(); in RemoveDecorationsFrom()
351 if (!module_) return; in AnalyzeDecorations()
354 for (Instruction& inst : module_->annotations()) { in AnalyzeDecorations()
390 IRContext* ctx = module_->context(); in AddDecoration()
513 auto context = module_->context(); in CloneDecorations()
516 std::unique_ptr<Instruction> new_inst(inst->Clone(module_->context())); in CloneDecorations()
518 module_->AddAnnotationInst(std::move(new_inst)); in CloneDecorations()
[all …]
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dir_loader.cpp37 module_(m), in IrLoader()
152 module_->AddFunction(std::move(function_)); in AddInstruction()
185 module_->AddCapability(std::move(spv_inst)); in AddInstruction()
187 module_->AddExtension(std::move(spv_inst)); in AddInstruction()
189 module_->AddExtInstImport(std::move(spv_inst)); in AddInstruction()
191 module_->SetMemoryModel(std::move(spv_inst)); in AddInstruction()
193 module_->SetSampledImageAddressMode(std::move(spv_inst)); in AddInstruction()
195 module_->AddEntryPoint(std::move(spv_inst)); in AddInstruction()
198 module_->AddExecutionMode(std::move(spv_inst)); in AddInstruction()
200 module_->AddDebug1Inst(std::move(spv_inst)); in AddInstruction()
[all …]
Ddecoration_manager.cpp64 auto context = module_->context(); in RemoveDecorationsFrom()
140 decoration->Clone(module_->context())); in RemoveDecorationsFrom()
142 module_->AddAnnotationInst(std::move(new_inst)); in RemoveDecorationsFrom()
143 auto decoration_iter = --module_->annotation_end(); in RemoveDecorationsFrom()
351 if (!module_) return; in AnalyzeDecorations()
354 for (Instruction& inst : module_->annotations()) { in AnalyzeDecorations()
390 IRContext* ctx = module_->context(); in AddDecoration()
513 auto context = module_->context(); in CloneDecorations()
516 std::unique_ptr<Instruction> new_inst(inst->Clone(module_->context())); in CloneDecorations()
518 module_->AddAnnotationInst(std::move(new_inst)); in CloneDecorations()
[all …]
Dir_loader.h49 Module* module() const { return module_; } in module()
54 module_->SetHeader({magic, version, generator, bound, reserved}); in SetModuleHeader()
74 Module* module_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc32 : module_(CreateNewVerifiedModule()), in WhileTransformerTest()
105 module_->AddEntryComputation(builder.Build()); in BuildWhileInstruction()
119 std::unique_ptr<HloModule> module_; member in xla::__anon29d942b10111::WhileTransformerTest
128 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F()
129 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1)); in TEST_F()
139 module_->AddEmbeddedComputation(BuildConditionComputation(1, 10)); in TEST_F()
140 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(1, 0, 1)); in TEST_F()
150 module_->AddEmbeddedComputation(BuildConditionComputation(0, 5)); in TEST_F()
151 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1)); in TEST_F()
161 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F()
[all …]
Dir_emitter_nested.cc74 llvm_ir::ShapeToIrType(param_shape, module_)->getPointerTo()); in CodegenNestedComputation()
76 llvm_ir::ByteSizeOf(param_shape, module_->getDataLayout()); in CodegenNestedComputation()
84 llvm_ir::ShapeToIrType(root_shape, module_)->getPointerTo()); in CodegenNestedComputation()
144 Load(llvm_ir::ShapeToIrType(return_shape, module_), root_value, in CodegenNestedComputation()
150 llvm::Type* tuple_type = llvm_ir::ShapeToIrType(return_shape, module_); in CodegenNestedComputation()
164 llvm_ir::ShapeToIrType(root_instruction->shape(), module_), &b_); in CodegenNestedComputation()
165 Store(Load(llvm_ir::ShapeToIrType(element_shape, module_), source), in CodegenNestedComputation()
207 ? llvm_ir::ConvertLiteralToIrConstant(literal, module_) in EmitConstants()
/external/google-breakpad/src/processor/
Dminidump.cc1848 module_(), in MinidumpModule()
1877 if (!minidump_->ReadBytes(&module_, MD_MODULE_SIZE)) { in Read()
1883 Swap(&module_.base_of_image); in Read()
1884 Swap(&module_.size_of_image); in Read()
1885 Swap(&module_.checksum); in Read()
1886 Swap(&module_.time_date_stamp); in Read()
1887 Swap(&module_.module_name_rva); in Read()
1888 Swap(&module_.version_info.signature); in Read()
1889 Swap(&module_.version_info.struct_version); in Read()
1890 Swap(&module_.version_info.file_version_hi); in Read()
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
Dhlo_extractor.cc47 module_(std::make_unique<HloModule>("extracted", config_)), in ExtractionVisitor()
48 clone_context_(module_.get()), in ExtractionVisitor()
83 module_->AddEntryComputation(builder_.Build()); in FinishVisit()
98 HloModule* module() { return module_.get(); } in module()
100 std::unique_ptr<HloModule> ConsumeModule() { return std::move(module_); } in ConsumeModule()
105 std::unique_ptr<HloModule> module_; member in xla::__anon1b44898a0111::ExtractionVisitor
/external/python/pybind11/include/pybind11/
Dembed.h49 static ::pybind11::module_::module_def \
51 static void PYBIND11_CONCAT(pybind11_init_, name)(::pybind11::module_ &); \
53 auto m = ::pybind11::module_::create_extension_module( \
65 void PYBIND11_CONCAT(pybind11_init_, name)(::pybind11::module_ &variable)
110 module_::import("sys").attr("path").cast<list>().append("."); in PYBIND11_NAMESPACE_END()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_module_importer.cc37 module_(module), in HloModuleImporter()
45 module_.setName(module.name()); in Import()
50 module_, &function_map_, &builder_, in Import()
56 *computation, module_, &function_map_, &builder_, in Import()
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Dtac_pass.h40 module_(module) {} in TacPass()
45 return module_ != nullptr in GetTargetHardware()
46 ? module_->GetTargetHardware(hardware_name) in GetTargetHardware()
51 const TacModule* module_ = nullptr; // Not owned.
/external/python/pybind11/tests/
Dpybind11_tests.cpp29 std::list<std::function<void(py::module_ &)>> &initializers() { in initializers()
30 static std::list<std::function<void(py::module_ &)>> inits; in initializers()
39 initializers().emplace_back([=](py::module_ &parent) { in test_initializer()
45 void bind_ConstructorStats(py::module_ &m) { in bind_ConstructorStats()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dalias_analysis.cc39 hlo.parent() == module_.entry_computation()) { in AddAliasingInformationToIrArray()
54 if (module_.config().debug_options().xla_llvm_enable_alias_scope_metadata()) { in AddAliasingInformationToIrArray()
65 if (module_.config().debug_options().xla_llvm_enable_noalias_metadata()) { in AddAliasingInformationToIrArray()
76 if (module_.config() in AddAliasingInformationToIrArray()
82 hlo.parent() == module_.entry_computation()) { in AddAliasingInformationToIrArray()

1234567