Searched defs:arg_shape (Results 1 – 2 of 2) sorted by relevance
229 py::tuple arg_shape = py_args[i].attr("shape"); in check_bprop_input_grads() local
642 auto arg_shape = arg->GetShape()->GetShapeVector(); in GetShapeValue() local