Searched refs:py_dtype (Results 1 – 3 of 3) sorted by relevance
43 Safe_PyObjectPtr py_dtype(PyObject_GetAttr(tensor, dtype_attr)); in DataTypeForTensor() local44 if (!py_dtype) return DT_INVALID; in DataTypeForTensor()46 Safe_PyObjectPtr enum_field(PyObject_GetAttr(py_dtype.get(), type_enum_attr)); in DataTypeForTensor()
115 Safe_PyObjectPtr py_dtype = GetAttr_DType(value); in InferDType() local116 if (!py_dtype) return false; in InferDType()117 dtype = DataTypeFromPyDType(py_dtype.get()); // set output parameter in InferDType()254 PyObject* py_dtype = params[input.type_attr->index]; in ConvertInputsWithTypeAttr() local255 dtype = DataTypeFromPyDType(py_dtype); in ConvertInputsWithTypeAttr()
480 const string py_dtype = in GenerateTypeVars() local482 allowed_types.emplace_back(dtype_type.at(py_dtype)); in GenerateTypeVars()1014 const string py_dtype = in AddEagerInferredAttrs() local1016 strings::StrAppend(&conversion, py_dtype, ", "); in AddEagerInferredAttrs()1286 const string py_dtype = in GetArgAnnotation() local1288 return "_ops.Tensor[" + dtype_type.at(py_dtype) + "]"; in GetArgAnnotation()