Searched refs:tensor_attr (Results 1 – 2 of 2) sorted by relevance
1340 TensorProto* tensor_attr = in TestConvertConst() local1342 tensor_attr->Clear(); in TestConvertConst()1345 tensor.AsProtoTensorContent(tensor_attr); in TestConvertConst()1347 tensor.shape().AsProto(tensor_attr->mutable_tensor_shape()); in TestConvertConst()1348 tensor_attr->set_dtype(tensor.dtype()); in TestConvertConst()1351 CopyTensorElements<float>(tensor, tensor_attr->mutable_float_val()); in TestConvertConst()1353 CopyTensorElements<int32>(tensor, tensor_attr->mutable_int_val()); in TestConvertConst()1355 tensor.AsProtoField(tensor_attr); in TestConvertConst()
1728 AttrValue tensor_attr; in ReplaceOperationWithConstant() local1730 CreateConstantTensorAttrValue(dtype, value, shape, &tensor_attr)); in ReplaceOperationWithConstant()1734 node->mutable_attr()->insert({"value", tensor_attr}); in ReplaceOperationWithConstant()