forked from apache/arrow
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtensor-dim.ts
More file actions
69 lines (58 loc) · 2.07 KB
/
Copy pathtensor-dim.ts
File metadata and controls
69 lines (58 loc) · 2.07 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
// automatically generated by the FlatBuffers compiler, do not modify
import * as flatbuffers from 'flatbuffers';
/**
* ----------------------------------------------------------------------
* Data structures for dense tensors
* Shape data for a single axis in a tensor
*/
export class TensorDim {
bb: flatbuffers.ByteBuffer|null = null;
bb_pos = 0;
__init(i:number, bb:flatbuffers.ByteBuffer):TensorDim {
this.bb_pos = i;
this.bb = bb;
return this;
}
static getRootAsTensorDim(bb:flatbuffers.ByteBuffer, obj?:TensorDim):TensorDim {
return (obj || new TensorDim()).__init(bb.readInt32(bb.position()) + bb.position(), bb);
}
static getSizePrefixedRootAsTensorDim(bb:flatbuffers.ByteBuffer, obj?:TensorDim):TensorDim {
bb.setPosition(bb.position() + flatbuffers.SIZE_PREFIX_LENGTH);
return (obj || new TensorDim()).__init(bb.readInt32(bb.position()) + bb.position(), bb);
}
/**
* Length of dimension
*/
size():bigint {
const offset = this.bb!.__offset(this.bb_pos, 4);
return offset ? this.bb!.readInt64(this.bb_pos + offset) : BigInt('0');
}
/**
* Name of the dimension, optional
*/
name():string|null
name(optionalEncoding:flatbuffers.Encoding):string|Uint8Array|null
name(optionalEncoding?:any):string|Uint8Array|null {
const offset = this.bb!.__offset(this.bb_pos, 6);
return offset ? this.bb!.__string(this.bb_pos + offset, optionalEncoding) : null;
}
static startTensorDim(builder:flatbuffers.Builder) {
builder.startObject(2);
}
static addSize(builder:flatbuffers.Builder, size:bigint) {
builder.addFieldInt64(0, size, BigInt('0'));
}
static addName(builder:flatbuffers.Builder, nameOffset:flatbuffers.Offset) {
builder.addFieldOffset(1, nameOffset, 0);
}
static endTensorDim(builder:flatbuffers.Builder):flatbuffers.Offset {
const offset = builder.endObject();
return offset;
}
static createTensorDim(builder:flatbuffers.Builder, size:bigint, nameOffset:flatbuffers.Offset):flatbuffers.Offset {
TensorDim.startTensorDim(builder);
TensorDim.addSize(builder, size);
TensorDim.addName(builder, nameOffset);
return TensorDim.endTensorDim(builder);
}
}