Searched refs:opBuilder (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
| D | OperationBuilderTest.java | 53 g.opBuilder("Const", "Const").setAttr("dtype", t.dataType()).setAttr("value", t); in failOnUseAfterBuild()
68 b = g.opBuilder("Const", "Const").setAttr("dtype", t.dataType()).setAttr("value", t); in failOnUseAfterGraphClose()
89 g.opBuilder("Const", "DataTypeAndTensor") in setAttr()
97 g.opBuilder("Abort", "StringAndBool") in setAttr()
103 g.opBuilder("RandomUniform", "Int") in setAttr()
110 g.opBuilder("MaxPool", "IntList") in setAttr()
118 g.opBuilder("FractionalMaxPool", "FloatList") in setAttr()
131 g.opBuilder("Placeholder", "unknown") in setAttrShape()
139 n = g.opBuilder("Placeholder", "batch_of_vectors") in setAttrShape()
172 g.opBuilder("Assert", "assert") in addControlInput()
[all …]
|
| D | TestUtil.java | 49 return g.opBuilder("Const", name) in constant()
58 return g.opBuilder("Placeholder", name) in placeholder()
65 return g.opBuilder("AddN", "AddN").addInputList(inputs).build().output(0); in addN()
70 return g.opBuilder("MatMul", name) in matmul()
80 return g.opBuilder("Split", name) in split()
88 return g.opBuilder("Square", name) in square()
|
| D | TensorFlowTest.java | 48 g.opBuilder("MyTest", "MyTest").build(); in loadLibrary()
59 g.opBuilder("MyTest", "MyTest").build(); in loadLibrary()
|
| D | OperationTest.java | 39 g.opBuilder("Add", "Add") in outputListLengthFailsOnInvalidName()
171 return g.opBuilder("Split", "Split") in split()
182 return g.opBuilder("Split", "Split") in splitWithInputList()
|
| D | SessionTest.java | 68 g.opBuilder("Split", "Split") in runUsingColonSeparatedNames()
73 g.opBuilder("Add", "Add") in runUsingColonSeparatedNames()
|
| D | GraphTest.java | 275 .opBuilder("Less", "cond") in buildWhileLoopSingleInput()
324 .opBuilder("Less", "cond") in buildWhileLoopMultipleInputs()
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/examples/ |
| D | LabelImage.java | 186 return g.opBuilder("Cast", "Cast") in cast()
194 return g.opBuilder("DecodeJpeg", "DecodeJpeg") in decodeJpeg()
203 return g.opBuilder("Const", name) in constant()
227 return g.opBuilder(type, type).addInput(in1).addInput(in2).build().<T>output(0); in binaryOp()
231 return g.opBuilder(type, type).addInput(in1).addInput(in2).build().<T>output(0); in binaryOp3()
|
| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/op/ |
| D | PrimitiveOpTest.java | 42 new PrimitiveOp(g.opBuilder("Shape", "shape1").addInput(array).build()) {}; in equalsHashcode()
44 new PrimitiveOp(g.opBuilder("Shape", "shape2").addInput(array).build()) {}; in equalsHashcode()
|
| D | ScopeTest.java | 200 .opBuilder("Const", s.makeOpName("Const")) in create()
211 .opBuilder("Const", s.makeOpName("Const")) in create()
234 .opBuilder("Mean", s.makeOpName("Mean")) in create()
256 .opBuilder("SquaredDifference", s.makeOpName("SquaredDifference")) in create()
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/ |
| D | Graph.java | 100 public OperationBuilder opBuilder(String type, String name) { in opBuilder() method in Graph
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/op/core/ |
| D | Constant.java | 644 .opBuilder("Const", scope.makeOpName("Const")) in createWithTensor()
|