Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_build_info.cc23 if (input_index >= inputs_format_.size()) { in GetInputFormat()
27 return inputs_format_[input_index]; in GetInputFormat()
56 const std::vector<std::string> &KernelBuildInfo::GetAllInputFormats() const { return inputs_format_ in GetAllInputFormats()
64 size_t KernelBuildInfo::GetInputNum() const { return inputs_format_.size(); } in GetInputNum()
122 if (inputs_format_ != other.inputs_format_ || outputs_format_ != other.outputs_format_) { in IsSimilarityKernelBuildInfo()
158 kernel_build_info_->inputs_format_ = inputs_format; in SetInputsFormat()
215 auto index_limit = kernel_build_info_->inputs_format_.size(); in SetInputFormat()
220 kernel_build_info_->inputs_format_[index] = format; in SetInputFormat()
Dkernel_build_info.h42 inputs_format_ = {}; in KernelBuildInfo()
110 std::vector<std::string> inputs_format_; variable
136 kernel_build_info_->inputs_format_.emplace_back(kernel_build_info->GetInputFormat(index)); in KernelBuildInfoBuilder()