| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | strategy.h | 41 : stage_(stage), inputs_(std::move(inputs)), internal_size_(0), internal_stragies_() {} in Strategy()
43 Strategy(const Strategy &another_stra) : stage_(another_stra.GetInputStage()) { in Strategy()
56 int64_t GetInputStage() const { return stage_; } in GetInputStage()
71 if ((stage_ != another_stra->GetInputStage()) || (inputs_ != another_stra->GetInputDim())) { in IsEqual()
84 const int64_t stage_;
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_pass_manager.h | 32 : PassManager(name, true), stage_(stage), flags_(context::GraphKernelFlags::GetInstance()) {} in GraphKernelPassManager()
44 size_t stage_; variable
|
| D | graph_kernel_pass_manager.cc | 35 … std::to_string(this->stage_) + "." + std::to_string(pass_id)) != pass_list.end() || in AddPass()
51 …return "stage" + std::to_string(stage_) + "_" + name() + "_" + std::to_string(pass_id) + "_" + pas… in GetPassFullname()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | bn_fp16_grad.cc | 88 int stage = stage_; in DoExecute()
165 stage_ = 0; in Run()
176 stage_ = static_cast<int>(stage); in Run()
|
| D | bn_fp16_grad.h | 39 int stage_ = 0; variable
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 219 if (graph->stage() == stage_ && user_node_graph->stage() == -1) { in Coloring()
426 if (graph == root_ || graph->stage() == -1 || !parameter_stage.count(stage_)) { in HandleSharedParameter()
435 if (stage_ == *parameter_stage.begin()) { in HandleSharedParameter()
436 if (graph->stage() == stage_) { in HandleSharedParameter()
442 auto send_out = InsertSend(parameter, user_stage, stage_, micro); in HandleSharedParameter()
449 …auto recv = InsertReceive(main_graph_, parameter, node, user.second, stage_, *parameter_stage.begi… in HandleSharedParameter()
482 if (*parameter_stage.begin() == stage_ && !virtual_param_ && requires_grad) { in ParameterColoring()
796 if (stage_ == user_stage) { in HandleParameterGraph()
808 auto send_out = InsertSend(argument, user_stage, stage_, micro); in HandleParameterGraph()
835 if (node_stage != stage_ && user_node_stage != stage_) { in CutBorder()
[all …]
|
| D | pipeline_transformer.h | 46 stage_(stage), in PipelineTransformer()
89 int64_t stage_; variable
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | bn_grad.cc | 88 int stage = stage_; in Execute()
178 stage_ = 0; in Run()
190 stage_ = static_cast<int>(stage); in Run()
|
| D | bn_grad.h | 38 int stage_ = 0; variable
|
| D | sparse_softmax_cross_entropy_with_logits.h | 52 int stage_; variable
|
| D | sparse_softmax_cross_entropy_with_logits.cc | 98 switch (stage_) { in Execute()
151 stage_ = stage; in Run()
|
| /third_party/mindspore/tests/ut/cpp/ir/ |
| D | manager_test.cc | 93 NestingSpecs(Stage *stage, std::string specs) : stage_(stage) { ParseSpecs(specs); } in NestingSpecs()
99 if (stage_->subs().find(name) != stage_->subs().end()) { in Name()
100 return stage_->subs()[name]; in Name()
200 Stage *stage_; member in mindspore::NestingSpecs
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | func_graph.h | 399 int64_t stage() { return stage_; } in stage()
400 void set_stage(int64_t stage) { stage_ = stage; } in set_stage()
488 int64_t stage_; variable
|
| D | anf.h | 109 stage_(-1), in AnfNode()
203 int64_t stage() { return stage_; } in stage()
204 void set_stage(const int &stage) { stage_ = stage; } in set_stage()
229 int64_t stage_; variable
|
| D | func_graph.cc | 51 stage_(-1) { in FuncGraph()
|
| /third_party/vk-gl-cts/external/vulkancts/modules/vulkan/util/ |
| D | vktDrawUtil.hpp | 130 VulkanShader (const vk::VkShaderStageFlagBits stage_, const vk::ProgramBinary& binary_);
|
| D | vktDrawUtil.cpp | 181 VulkanShader::VulkanShader(const vk::VkShaderStageFlagBits stage_, const vk::ProgramBinary& binary_) in VulkanShader() argument
182 : stage(stage_) in VulkanShader()
|
| /third_party/skia/third_party/externals/swiftshader/include/vulkan/ |
| D | vulkan_structs.hpp | 10478 … VULKAN_HPP_CONSTEXPR CheckpointData2NV( VULKAN_HPP_NAMESPACE::PipelineStageFlags2KHR stage_ = {}, in CheckpointData2NV()
10480 : stage( stage_ )
10552 …VULKAN_HPP_NAMESPACE::PipelineStageFlagBits stage_ = VULKAN_HPP_NAMESPACE::PipelineStageFlagBits::… in CheckpointDataNV()
10554 : stage( stage_ )
12346 …VULKAN_HPP_NAMESPACE::ShaderStageFlagBits stage_ = VULKAN_HPP_NAMESPACE::ShaderStageFl…
12351 , stage( stage_ )
12388 …PipelineShaderStageCreateInfo & setStage( VULKAN_HPP_NAMESPACE::ShaderStageFlagBits stage_ ) VULKA… in setStage()
12390 stage = stage_; in setStage()
12472 … VULKAN_HPP_NAMESPACE::PipelineShaderStageCreateInfo stage_ = {},
12477 , stage( stage_ )
[all …]
|
| /third_party/vulkan-headers/include/vulkan/ |
| D | vulkan_structs.hpp | 10708 … VULKAN_HPP_CONSTEXPR CheckpointData2NV( VULKAN_HPP_NAMESPACE::PipelineStageFlags2KHR stage_ = {}, in CheckpointData2NV()
10710 : stage( stage_ )
10782 …VULKAN_HPP_NAMESPACE::PipelineStageFlagBits stage_ = VULKAN_HPP_NAMESPACE::PipelineStageFlagBits::… in CheckpointDataNV()
10784 : stage( stage_ )
12762 …VULKAN_HPP_NAMESPACE::ShaderStageFlagBits stage_ = VULKAN_HPP_NAMESPACE::ShaderStageFl…
12767 , stage( stage_ )
12805 … setStage( VULKAN_HPP_NAMESPACE::ShaderStageFlagBits stage_ ) VULKAN_HPP_NOEXCEPT in setStage()
12807 stage = stage_; in setStage()
12890 … VULKAN_HPP_NAMESPACE::PipelineShaderStageCreateInfo stage_ = {},
12895 , stage( stage_ )
[all …]
|