| /external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
| D | EagerOperationBuilderTest.java | 42 EagerOperationBuilder opBuilder; in failToBuildOpIfSessionIsClosed() local
44 opBuilder = new EagerOperationBuilder(session, "Empty", "empty"); in failToBuildOpIfSessionIsClosed()
47 opBuilder.setAttr("dtype", DataType.FLOAT); in failToBuildOpIfSessionIsClosed()
58 opBuilder(session, "Assert", "assert") in addInputs()
63 opBuilder(session, "Const", "var").addControlInput(asrt); in addInputs()
74 opBuilder(session, "Add", "SetDevice") in setDevice()
93 opBuilder(session, "Const", "DataTypeAndTensor") in setAttrs()
99 opBuilder(session, "RandomUniform", "DataTypeAndInt") in setAttrs()
105 opBuilder(session, "MaxPool", "IntListAndString") in setAttrs()
112 opBuilder(session, "FractionalMaxPool", "FloatList") in setAttrs()
[all …]
|
| D | GraphOperationBuilderTest.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()
140 g.opBuilder("Placeholder", "batch_of_vectors") in setAttrShape()
173 g.opBuilder("Assert", "assert") in addControlInput()
[all …]
|
| D | EagerOperationTest.java | 46 opBuilder(session, "Const", "OutputAttrs") in outputDataTypeAndShape()
60 opBuilder(session, "Add", "CompareResult") in outputTensor()
80 opBuilder(session, "AddN", "InputListLength") in inputAndOutputListLengths()
87 opBuilder(session, "Split", "OutputListLength") in inputAndOutputListLengths()
115 opBuilder(session, "UniqueWithCountsV2", "unq") in numOutputs()
128 opBuilder(session, "Add", "SessionClosed") in opNotAccessibleIfSessionIsClosed()
146 opBuilder(session, "Add", "OutOfRange") in outputIndexOutOfBounds()
177 private static EagerOperationBuilder opBuilder(EagerSession session, String type, String name) { in opBuilder() method in EagerOperationTest
|
| D | TestUtil.java | 49 return g.opBuilder("Const", name).setAttr("dtype", t.dataType()).setAttr("value", t).build(); in constantOp()
55 return env.opBuilder("Const", name) in constant()
64 return g.opBuilder("Placeholder", name) in placeholder()
71 return env.opBuilder("AddN", "AddN").addInputList(inputs).build().output(0); in addN()
76 return g.opBuilder("MatMul", name) in matmul()
86 return g.opBuilder("Split", name) in split()
94 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 | GraphOperationTest.java | 40 g.opBuilder("Add", "Add") in outputListLengthFailsOnInvalidName()
184 return g.opBuilder("Split", "Split") in split()
195 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()
|
| D | EagerSessionTest.java | 209 s.opBuilder("Const", "Const"); in buildOp()
|
| /external/llvm-project/mlir/lib/Dialect/SPIRV/Serialization/ |
| D | Deserializer.cpp | 310 Location createFileLineColLoc(OpBuilder opBuilder);
488 OpBuilder opBuilder; member in __anonf1899d380111::Deserializer
585 module(createModuleOp()), opBuilder(module->body()) {} in Deserializer()
735 opBuilder.getI32IntegerAttr(llvm::bit_cast<int32_t>(operands.front()))); in processMemoryModel()
738 opBuilder.getI32IntegerAttr(llvm::bit_cast<int32_t>(operands.back()))); in processMemoryModel()
757 auto symbol = opBuilder.getIdentifier(attrName); in processDecoration()
766 symbol, opBuilder.getI32IntegerAttr(static_cast<int32_t>(words[2]))); in processDecoration()
774 symbol, opBuilder.getStringAttr( in processDecoration()
800 decorations[words[0]].set(symbol, opBuilder.getUnitAttr()); in processDecoration()
809 symbol, opBuilder.getI32IntegerAttr(static_cast<int32_t>(words[2]))); in processDecoration()
[all …]
|
| /external/llvm-project/mlir/unittests/Dialect/SPIRV/ |
| D | SerializationTest.cpp | 63 OpBuilder opBuilder(module->body()); in getFloatStructType() local
64 llvm::SmallVector<Type, 1> elementTypes{opBuilder.getF32Type()}; in getFloatStructType()
71 OpBuilder opBuilder(module->body()); in addGlobalVar() local
73 opBuilder.create<spirv::GlobalVariableOp>( in addGlobalVar()
75 opBuilder.getStringAttr(name), nullptr); in addGlobalVar()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/Transforms/ |
| D | AffineLoopNormalize.cpp | 105 OpBuilder opBuilder(op); in normalizeAffineFor() local
154 newUbExprs, opBuilder.getContext()); in normalizeAffineFor()
158 op.setLowerBound({}, opBuilder.getConstantAffineMap(0)); in normalizeAffineFor()
163 opBuilder.setInsertionPointToStart(op.getBody()); in normalizeAffineFor()
173 AffineExpr origIVExpr = opBuilder.getAffineDimExpr(lb.getMap().getNumDims()); in normalizeAffineFor()
177 Operation *newIV = opBuilder.create<AffineApplyOp>(loc, ivMap, lbOperands); in normalizeAffineFor()
|
| /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 | 201 .opBuilder("Const", s.makeOpName("Const")) in create()
212 .opBuilder("Const", s.makeOpName("Const")) in create()
235 .opBuilder("Mean", s.makeOpName("Mean")) in create()
257 .opBuilder("SquaredDifference", s.makeOpName("SquaredDifference")) in create()
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/ |
| D | ExecutionEnvironment.java | 30 OperationBuilder opBuilder(String type, String name); in opBuilder() method
|
| D | EagerSession.java | 339 public OperationBuilder opBuilder(String type, String name) { in opBuilder() method in EagerSession
|
| D | Graph.java | 101 public GraphOperationBuilder opBuilder(String type, String name) { in opBuilder() method in Graph
|
| /external/llvm-project/mlir/lib/Parser/ |
| D | Parser.cpp | 107 : Parser(state), opBuilder(topLevelOp->getRegion(0)), in OperationParser()
314 OpBuilder opBuilder; member in __anond23dd3090111::OperationParser
851 Operation *op = opBuilder.createOperation(result); in parseGenericOperation()
859 OpBuilder::InsertionGuard restoreInsertionPoint(opBuilder); in parseGenericOperation()
860 opBuilder.setInsertionPoint(insertBlock, insertPt); in parseGenericOperation()
1657 Operation *op = opBuilder.createOperation(opState); in parseCustomOperation()
1727 auto currentPt = opBuilder.saveInsertionPoint(); in parseRegion()
1778 opBuilder.restoreInsertionPoint(currentPt); in parseRegion()
1841 opBuilder.setInsertionPointToEnd(block); in parseBlockBody()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 92 OpBuilder &opBuilder, OperationState &result, TypeRange inputTypes,
1454 OpBuilder &opBuilder, Region ®ion, TypeRange inputTypes, in buildNamedStructuredOpRegionAndAttributesImpl() argument
1466 OpBuilder::InsertionGuard guard(opBuilder); in buildNamedStructuredOpRegionAndAttributesImpl()
1467 Block *body = opBuilder.createBlock(®ion, {}, argTypes); in buildNamedStructuredOpRegionAndAttributesImpl()
1473 opBuilder.setInsertionPointToStart(body); in buildNamedStructuredOpRegionAndAttributesImpl()
1474 mlir::edsc::ScopedContext scope(opBuilder, opBuilder.getUnknownLoc()); in buildNamedStructuredOpRegionAndAttributesImpl()
1483 void buildNamedStructuredOpRegionAndAttributes(OpBuilder &opBuilder, in buildNamedStructuredOpRegionAndAttributes() argument
1491 opBuilder, region, inputTypes, outputBufferTypes, initTensorTypes, in buildNamedStructuredOpRegionAndAttributes()
1505 OpBuilder opBuilder(parser.getBuilder().getContext()); in parseNamedStructuredOpRegion() local
1507 opBuilder, region, inputTypes, outputBufferTypes, initTensorTypes, in parseNamedStructuredOpRegion()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/EDSC/ |
| D | Builders.cpp | 89 OpBuilder opBuilder(op); in makeGenericLinalgOp() local
90 ScopedContext scope(opBuilder, op->getLoc()); in makeGenericLinalgOp()
|
| /external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/op/core/ |
| D | Constant.java | 644 .opBuilder("Const", scope.makeOpName("Const")) in createWithTensor()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Operation.cpp | 1244 OpBuilder opBuilder(builder.getContext()); in ensureRegionTerminator() local
1245 ensureRegionTerminator(region, opBuilder, loc, buildTerminatorOp); in ensureRegionTerminator()
|