Searched refs:kInputNum (Results 1 – 7 of 7) sorted by relevance
23 const int kInputNum = 2; variable27 for (size_t i = 0; i < kInputNum; ++i) { in UpdatePreIsOne()35 for (size_t i = 0; i < kInputNum; ++i) { in AddElementToGradReduceIdx()44 std::vector<std::vector<int64_t>> grad_reduce_idx(kInputNum); in GetGradientIndices()46 std::vector<bool> prev_is_one(kInputNum); in GetGradientIndices()47 std::vector<bool> current_is_one(kInputNum); in GetGradientIndices()48 for (size_t i = 0; i < kInputNum; ++i) { in GetGradientIndices()59 for (size_t i = 0; i < kInputNum; ++i) { in GetGradientIndices()75 for (size_t i = 0; i < kInputNum; ++i) { in GetGradientIndices()91 std::vector<std::vector<int64_t>> grad_reduce_idx(kInputNum); in CalculateOutput()[all …]
35 constexpr size_t kInputNum = 4; variable42 …if ((in_tensors_.size() != kInputNum && in_tensors_.size() != kInputNumOpt) || out_tensors_.size()… in Init()43 … MS_LOG(ERROR) << "OneHot input size should be " << kInputNum << " or " << kInputNumOpt << ", got " in Init()154 if (in_tensors_.size() == kInputNum) { in InitParamsAndOnOffValue()
31 constexpr size_t kInputNum = 1; variable36 if (in_tensors_.size() < kInputNum) { in CheckInputsOutputs()37 …MS_LOG(ERROR) << "Reduce inputs size should be at least " << kInputNum << " but got " << in_tensor… in CheckInputsOutputs()
33 constexpr int kInputNum = 2; variable42 if (in_tensors_.size() != kInputNum) { in Init()
23 constexpr size_t kInputNum = 1; variable27 if (input_tensors_.size() < kInputNum) { in CheckInputsOutputs()28 …MS_LOG(ERROR) << "Reduce inputs size should be at least " << kInputNum << " but got " << input_ten… in CheckInputsOutputs()
26 constexpr size_t kInputNum = 3; variable107 if (cnode->size() != kInputNum + 1) { in Process()108 …MS_LOG(INFO) << "The node " << cnode->DebugString() << " is not equal to " << kInputNum << " input… in Process()
1521 const auto kInputNum = 3; in ConvertReshape() local1522 if (node->size() < kInputNum) { in ConvertReshape()