Searched refs:output_num_value (Results 1 – 2 of 2) sorted by relevance
846 int64_t output_num_value = GetValue<int64_t>(primitive->GetAttr("output_num")); in InferImplSplit() local847 …if ((x_shape[axis_value_pos] != Shape::SHP_ANY) && (x_shape[axis_value_pos] % output_num_value != … in InferImplSplit()849 << " must be divisible by output_num = " << output_num_value; in InferImplSplit()854 output_shape[axis_value_pos] = static_cast<int>(x_shape[axis_value_pos] / output_num_value); in InferImplSplit()857 …utput_shape_min[axis_value_pos] = static_cast<int>(x_shape_min[axis_value_pos] / output_num_value); in InferImplSplit()859 …utput_shape_max[axis_value_pos] = static_cast<int>(x_shape_max[axis_value_pos] / output_num_value); in InferImplSplit()862 for (int64_t i = 0; i < output_num_value; ++i) { in InferImplSplit()
531 auto output_num_value = prim->GetAttr(kOutputNum); in CheckCustomPrimOutputInferResult() local532 if (output_num_value == nullptr) { in CheckCustomPrimOutputInferResult()536 int64_t output_num = GetValue<int64_t>(output_num_value); in CheckCustomPrimOutputInferResult()