Searched defs:dtype_attr (Results 1 – 10 of 10) sorted by relevance
31 const AttrValue* dtype_attr = context->GetAttr("T"); in __anonff0e774f0102() local
40 static PyObject* dtype_attr = PY_STRING_INTERN_FROM_STRING("dtype"); in DataTypeForTensor() local
61 def dtype_attr(self): member in _TensorData
112 AttrValue dtype_attr; in BuildConstNode() local211 AttrValue dtype_attr; in TEST() local465 AttrValue dtype_attr; in NodeWithTensorContent() local581 AttrValue dtype_attr; in BuildUnaryNode() local
55 TypeAttr dtype_attr = op->getAttrOfType<TypeAttr>("T"); in matchAndRewrite() local
2437 AttrValue dtype_attr; in CreateIdentityGraphDef() local2797 AttrValue dtype_attr; in Input() local
858 AttrValue dtype_attr; in UpdateFunction() local
103 } else if (auto dtype_attr = op->getAttrOfType<TypeAttr>("dtype")) { in GetDataTypeFromOp() local
1186 const AttrValue* dtype_attr = node.attrs().Find("_handle_dtypes"); in InferOutputType() local
1162 static PyObject* dtype_attr = PyString_InternFromString("dtype"); in PyTensor_DataType() local