Searched defs:py_dtype (Results 1 – 6 of 6) sorted by relevance
116 auto py_dtype = [](pybind11::dtype dtype) -> char { in ConvertPyArrayMemrefDesc() local
167 auto py_dtype = [](pybind11::dtype dtype) -> char { in ConvertPyArrayMemrefDesc() local
43 Safe_PyObjectPtr py_dtype(PyObject_GetAttr(tensor, dtype_attr)); in DataTypeForTensor() local
115 Safe_PyObjectPtr py_dtype = GetAttr_DType(value); in InferDType() local254 PyObject* py_dtype = params[input.type_attr->index]; in ConvertInputsWithTypeAttr() local
482 const string py_dtype = in GenerateTypeVars() local1023 const string py_dtype = in AddEagerInferredAttrs() local1321 const string py_dtype = in GetArgAnnotation() local
774 PyObject* py_dtype = PyTuple_GET_ITEM(args, 2); in _parse_empty_strided_args() local