forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathEagerOperation.cs
More file actions
81 lines (69 loc) · 2.08 KB
/
EagerOperation.cs
File metadata and controls
81 lines (69 loc) · 2.08 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using static Tensorflow.Binding;
namespace Tensorflow.Eager
{
public class EagerOperation : Operation
{
public string Name { get; set; }
public new int NumInputs;
public IntPtr[] InputHandles { get; set; }
public Tensor[] Inputs { get; set; }
public new int NumOutputs;
public IntPtr[] OutputHandles { get; set; }
public Tensor[] Outputs { get; set; }
public long[] SkipInputIndices { get; set; }
public object[] Attrs { get; set; }
public EagerOperation() : base(IntPtr.Zero)
{
}
public override InputList inputs
{
get
{
if (_inputs_val == null)
{
_inputs_val = new InputList(Inputs);
}
return _inputs_val;
}
}
public override Tensor[] outputs
{
get
{
if (_outputs == null)
{
_outputs = Outputs;
}
return _outputs;
}
}
public override object get_attr(string attr_name)
{
object value = null;
byte isList = 0;
var attrType = c_api.TFE_OpNameGetAttrType(tf.context.Handle, Name, attr_name, ref isList, tf.status.Handle);
switch (attrType)
{
case TF_AttrType.TF_ATTR_BOOL:
value = get_attr_bool(attr_name);
break;
default:
break;
}
return value;
}
public bool get_attr_bool(string attr_name)
{
for (int i = 0; i < Attrs.Length; i = i + 2)
if (Attrs[i].Equals(attr_name))
return Attrs[i + 1].Equals("1");
throw new ValueError($"Can't find attr: {attr_name}");
}
public override string ToString()
=> $"tf.EagerOperation {Name}";
}
}