Searched defs:axis_ptr (Results 1 – 13 of 13) sorted by relevance
44 auto axis_ptr = input_args[1]->GetValue(); in InferShape() local124 auto axis_ptr = input_args[1]->GetValue(); in CheckValidation() local
37 auto axis_ptr = input_args[kIndex1]; in InferShape() local
39 auto axis_ptr = primitive->GetAttr(ops::kAxes); in Check() local
40 auto axis_ptr = primitive->GetAttr(ops::kAxis); in Check() local
51 auto axis_ptr = primitive->GetAttr(ops::kAxis); in Check() local
56 auto axis_ptr = primitive->GetAttr(ops::kAxis); in Check() local
47 auto axis_ptr = src_prim->GetAttr(ops::kAxis); in Mapper() local
56 auto axis_ptr = primitive->GetAttr(kAxis); in UniqueConsecutiveInferShape() local117 auto axis_ptr = primitive->GetAttr(kAxis); in UniqueConsecutiveFrontendInferShape() local
311 auto axis_ptr = primitive->GetAttr("axis"); in CheckAndGetAxisValueFromAttr() local
56 for (auto &axis_ptr : axis_ptr_list) { in InferImplReduceFuncCalShape() local
191 AttrValue *axis_ptr = ctx.GetAttr("axis"); in MedianGradCompute() local
151 auto axis_ptr = primitive_->GetAttr("axis"); in Init() local