Searched refs:input_value_ (Results 1 – 16 of 16) sorted by relevance
84 if (input_value_.size() != STRIDED_SLICE_INPUTS_SIZE) { in GetAttrs()86 << input_value_.size(); in GetAttrs()90 if ((GetInput(input_value_[STRIDED_SLICE_BEGIN_INDEX], &begin_) != SUCCESS) || in GetAttrs()91 (GetInput(input_value_[STRIDED_SLICE_END_INDEX], &end_) != SUCCESS) || in GetAttrs()92 (GetInput(input_value_[STRIDED_SLICE_STRIDES_INDEX], &strides_) != SUCCESS)) { in GetAttrs()
32 if (input_value_.size() != 3) { in GetAttrs()33 MS_LOG(ERROR) << name_ << ": Invalid input_value's size " << input_value_.size(); in GetAttrs()37 if (!input_value_[1]->isa<Int64Imm>()) { in GetAttrs()42 int64_t dim = GetValue<int64_t>(input_value_[1]); in GetAttrs()
55 if (input_value_.size() != SLICE_INPUTS_SIZE) { in GetAttrs()57 << input_value_.size(); in GetAttrs()61 if ((GetInput(input_value_[SLICE_BEGIN_INDEX], &begin_) != SUCCESS) || in GetAttrs()62 (GetInput(input_value_[SLICE_SIZE_INDEX], &size_) != SUCCESS)) { in GetAttrs()
33 if (input_value_.size() < 2) { in GetAttrs()37 if (input_value_[1] == nullptr) { in GetAttrs()43 ValueTuplePtr multiples = input_value_[1]->cast<ValueTuplePtr>(); in GetAttrs()
42 if (input_value_.size() != GATHER_V2_INPUTS_VALUE_SIZE) { in GetAttrs()43 MS_LOG(ERROR) << name_ << ": input value size must be 3, but is " << input_value_.size(); in GetAttrs()49 if (input_value_.at(2) == nullptr) { in GetAttrs()58 int64_t axis = GetValue<int64_t>(input_value_.at(2)); in GetAttrs()
42 if (input_value_.at(2) == nullptr) { in GetAttrs()51 auto num_segments = GetValue<int64_t>(input_value_.at(2)); in GetAttrs()280 auto num_segments = GetValue<int64_t>(input_value_.at(2)); in ComputeReplaceGraph()315 auto num_segments = GetValue<int64_t>(input_value_.at(2)); in ComputeReplaceGraph()
45 if (input_value_.size() < 2) { in reduce_dim()48 if (input_value_.back() == nullptr) { in reduce_dim()53 if (input_value_.back()->isa<ValueTuple>()) { in reduce_dim()54 auto attr_axis = GetValue<std::vector<int64_t>>(input_value_.back()); in reduce_dim()65 } else if (input_value_.back()->isa<Int64Imm>()) { in reduce_dim()66 int64_t axis = GetValue<int64_t>(input_value_.back()); in reduce_dim()
53 if (input_value_[1] == nullptr) { in ComputeAxis()58 ValueTuplePtr dim_tuple = input_value_[1]->cast<ValueTuplePtr>(); in ComputeAxis()
142 if (input_value_.size() != 4) { in ExtractInputInfo()145 << input_value_.size(); in ExtractInputInfo()148 auto value_ptr = input_value_.at(1); in ExtractInputInfo()
117 if (input_value_.empty()) { in GetAttrs()121 auto shape_ptr = input_value_[0]->cast<ValueTuplePtr>(); in GetAttrs()
383 if (input_value_.size() != EXPANDDIMS_INPUT_SIZE) { in GetAttrs()384 MS_LOG(ERROR) << name_ << ": Invalid inputs size " << input_value_.size(); in GetAttrs()388 if (!input_value_.back()->isa<Int64Imm>()) { in GetAttrs()393 int64_t axis = GetValue<int64_t>(input_value_.back()); in GetAttrs()
156 void set_input_value(const std::vector<ValuePtr> &input_value) { input_value_ = input_value; } in set_input_value()157 const std::vector<ValuePtr> &input_value() const { return input_value_; } in input_value()231 std::vector<ValuePtr> input_value_; variable
42 ValueTuplePtr shape_value = input_value_[0]->cast<ValueTuplePtr>(); in GetAttrs()
860 if (input_value_.size() != 3) { in GetOutShape()861 … MS_LOG(ERROR) << name_ << ": The size of input value must be 3, but got " << input_value_.size(); in GetOutShape()865 if (input_value_[2] == nullptr) { in GetOutShape()871 auto value_tuple = input_value_[2]->cast<ValueTuplePtr>(); in GetOutShape()
86 if (input_value_[1] == nullptr) { in GetParameterInput()91 ValueTuplePtr dim_tuple = input_value_[1]->cast<ValueTuplePtr>(); in GetParameterInput()
124 if (input_value_.at(2) == nullptr) { in GetAttrs()128 auto axis = GetValue<int64_t>(input_value_.at(2)); in GetAttrs()