Home
last modified time | relevance | path

Searched refs:outputs_format_ (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_build_info.cc31 if (output_index >= outputs_format_.size()) { in GetOutputFormat()
35 return outputs_format_[output_index]; in GetOutputFormat()
58 … std::vector<std::string> &KernelBuildInfo::GetAllOutputFormats() const { return outputs_format_; } in GetAllOutputFormats()
66 size_t KernelBuildInfo::GetOutputNum() const { return outputs_format_.size(); } in GetOutputNum()
122 if (inputs_format_ != other.inputs_format_ || outputs_format_ != other.outputs_format_) { in IsSimilarityKernelBuildInfo()
163 kernel_build_info_->outputs_format_ = outputs_format; in SetOutputsFormat()
225 auto index_limit = kernel_build_info_->outputs_format_.size(); in SetOutputFormat()
230 kernel_build_info_->outputs_format_[index] = format; in SetOutputFormat()
Dkernel_build_info.h43 outputs_format_ = {}; in KernelBuildInfo()
112 std::vector<std::string> outputs_format_; variable
142 kernel_build_info_->outputs_format_.emplace_back(kernel_build_info->GetOutputFormat(index)); in KernelBuildInfoBuilder()