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
63 lines (53 loc) · 1.56 KB
/
EagerOperation.cs
File metadata and controls
63 lines (53 loc) · 1.56 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
using System;
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)
{
// var attrType = c_api.TFE_OpNameGetAttrType(tf.Context.Handle, Name, attr_name, ref isList, tf.Status.Handle);
for (int i = 0; i < Attrs.Length; i = i + 2)
{
if (Attrs[i].Equals(attr_name))
return Attrs[i + 1];
}
return null;
}
public override string ToString()
=> $"tf.EagerOperation {Name}";
}
}