Searched defs:axis_vec (Results 1 – 17 of 17) sorted by relevance
39 auto axis_vec = value->axis(); in PopulateL2NormParameter() local
65 std::vector<int64_t> axis_vec = axis_array.ToVector(); in ReduceInferShape() local189 std::vector<int64_t> axis_vec = axis_array.ToVector(); in ReduceExtandInferShape() local233 std::vector<int64_t> axis_vec; in ReduceInferShape() local
42 auto axis_vec = GetValue<std::vector<int64_t>>(attr_axis); in PopulateL2NormOpParameter() local
30 auto axis_vec = transform::ConvertKernelTensor<std::vector<int64_t>>(axis_ptr); in GetDimValue() local
31 auto axis_vec = transform::ConvertKernelTensor<std::vector<int64_t>>(axis_ptr); in GetDimValue() local
32 auto axis_vec = transform::ConvertKernelTensor<std::vector<int64_t>>(axis_ptr); in GetDimValue() local
40 auto axis_vec = ConvertTensorAsIntVector(in_tensors_[AXIS_INDEX]); in IsSupport() local
115 auto axis_vec = ConvertTensorAsIntVector(axis_tensor); in GetAxis() local
51 auto axis_vec = ConvertTensorAsIntVector(in_tensors_[1]); in AddInnerOp() local
315 auto axis_vec = ConvertTensorAsIntVector(in_tensors_[1]); in AddExpandDimsOp() local
64 auto axis_vec = ConvertTensorAsIntVector(in_tensors[axis_index]); in GetConstInputValue() local
305 auto axis_vec = ConvertTensorAsIntVector(in_tensors_[axis_index]); in GetAxis() local
88 …auto axis_vec = CheckAndConvertUtils::CheckIntOrTupleInt("attribute[axis]", primitive->GetAttr("ax… in InferShape() local
76 auto axis_vec = inputs[kIndex1]->GetValueWithCheck<std::vector<int64_t>>(); in Resize() local
113 ShapeVector axis_vec; in Process() local
52 auto axis_vec = GetReduceAxis(node); in GetUniqReduceAxes() local
738 std::vector<int64_t> axis_vec; in CreateSqueezeNode() local