Searched refs:dtype_attr (Results 1 – 8 of 8) sorted by relevance
31 const AttrValue* dtype_attr = context->attrs().Find("T"); in __anon5bd79b600102() local32 if (!dtype_attr) { in __anon5bd79b600102()53 if (dtype_attr->type() != DT_RESOURCE) { in __anon5bd79b600102()59 dtype_attr = context->attrs().Find("_handle_dtypes"); in __anon5bd79b600102()63 if (!dtype_attr || !shape_attr) { in __anon5bd79b600102()67 if (dtype_attr->list().type().empty()) { in __anon5bd79b600102()77 DataType dtype = dtype_attr->list().type(0); in __anon5bd79b600102()
108 AttrValue dtype_attr; in BuildConstNode() local109 SetAttrValue(dtype, &dtype_attr); in BuildConstNode()110 (*node->mutable_attr())["dtype"] = dtype_attr; in BuildConstNode()202 AttrValue dtype_attr; in TEST() local203 SetAttrValue(0.5, &dtype_attr); in TEST()204 (*node.mutable_attr())["_softmax_beta"] = dtype_attr; in TEST()449 AttrValue dtype_attr; in NodeWithTensorContent() local450 SetAttrValue(dtype, &dtype_attr); in NodeWithTensorContent()451 (*node->mutable_attr())["dtype"] = dtype_attr; in NodeWithTensorContent()565 AttrValue dtype_attr; in BuildUnaryNode() local[all …]
40 static PyObject* dtype_attr = PY_STRING_INTERN_FROM_STRING("dtype"); in DataTypeForTensor() local43 Safe_PyObjectPtr py_dtype(PyObject_GetAttr(tensor, dtype_attr)); in DataTypeForTensor()
60 def dtype_attr(self): member in _TensorData388 node.attr["dtype"].CopyFrom(tensor_data.dtype_attr)
2404 AttrValue dtype_attr; in CreateIdentityGraphDef() local2405 dtype_attr.set_type(dtype); in CreateIdentityGraphDef()2413 placeholder->mutable_attr()->insert({"dtype", dtype_attr}); in CreateIdentityGraphDef()2420 identity->mutable_attr()->insert({"T", dtype_attr}); in CreateIdentityGraphDef()2764 AttrValue dtype_attr; in Input() local2765 SetAttrValue(DT_FLOAT, &dtype_attr); in Input()2769 input.mutable_attr()->insert({"dtype", dtype_attr}); in Input()
829 AttrValue dtype_attr; in UpdateFunction() local834 dtype_attr.mutable_list()->add_type(dtype); in UpdateFunction()838 (*fun_node->mutable_attr())["_handle_dtypes"] = dtype_attr; in UpdateFunction()
1039 const AttrValue* dtype_attr = node.attrs().Find("_handle_dtypes"); in InferOutputType() local1041 if (dtype_attr && shape_attr) { in InferOutputType()1042 if (dtype_attr->list().type().empty()) { in InferOutputType()1052 DataType dtype = dtype_attr->list().type(0); in InferOutputType()
1122 static PyObject* dtype_attr = PyString_InternFromString("dtype"); in PyTensor_DataType() local1126 static PyObject* dtype_attr = PyUnicode_InternFromString("dtype"); in PyTensor_DataType()1129 Safe_PyObjectPtr dtype_field(PyObject_GetAttr(tensor, dtype_attr)); in PyTensor_DataType()