blob: 976cd9fd347fb48020f186ece8d23dae89626b1d (
plain)
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
|
/* Copyright 2017 The TensorFlow Authors. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
package org.tensorflow.types;
import java.util.HashMap;
import java.util.Map;
import org.tensorflow.DataType;
/**
* Utility class for managing the representation of TensorFlow types as Java
* types. For each TensorFlow type (e.g., int32), there is a corresponding Java
* type (e.g., TFInt32) that represents it at compile time and a corresponding
* class object (e.g., TFInt32.class) that represents it at run time. There is
* also an enumeration value in DataType that can be used to represent the
* type, though that should rarely be required.
*/
public class Types {
private Types() {} // not instantiable
static final Map<Class<?>, DataType> typeCodes = new HashMap<>();
/** Returns the DataType value corresponding to a TensorFlow type class. */
public static DataType dataType(Class<? extends TFType> c) {
DataType dtype = typeCodes.get(c);
if (dtype == null) {
throw new IllegalArgumentException("" + c + " is not a TensorFlow type.");
}
return dtype;
}
static final Map<Class<?>, Object> scalars = new HashMap<>();
/** Returns the zero value of type described by {@code c}, or null if
* the type (e.g., string) is not numeric and therefore has no zero value.
*/
public static Object zeroValue(Class<? extends TFType> c) {
return scalars.get(c);
}
}
|
