|
@@ -380,8 +380,10 @@ namespace Tensorflow.Eager |
|
|
c_api.TFE_OpSetAttrBool(op, key, Convert.ToBoolean(value)); |
|
|
c_api.TFE_OpSetAttrBool(op, key, Convert.ToBoolean(value)); |
|
|
break; |
|
|
break; |
|
|
case TF_AttrType.TF_ATTR_INT: |
|
|
case TF_AttrType.TF_ATTR_INT: |
|
|
attr_list_sizes[key] = Convert.ToInt64(value); |
|
|
|
|
|
c_api.TFE_OpSetAttrInt(op, key, attr_list_sizes[key]); |
|
|
|
|
|
|
|
|
var size = Convert.ToInt64(value); |
|
|
|
|
|
c_api.TFE_OpSetAttrInt(op, key, size); |
|
|
|
|
|
if (attr_list_sizes != null) |
|
|
|
|
|
attr_list_sizes[key] = size; |
|
|
break; |
|
|
break; |
|
|
case TF_AttrType.TF_ATTR_FLOAT: |
|
|
case TF_AttrType.TF_ATTR_FLOAT: |
|
|
c_api.TFE_OpSetAttrFloat(op, key, Convert.ToSingle(value)); |
|
|
c_api.TFE_OpSetAttrFloat(op, key, Convert.ToSingle(value)); |
|
|