Home
last modified time | relevance | path

Searched refs:OperationBuilder (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DOperationBuilder.java34 public interface OperationBuilder { interface
57 public OperationBuilder addInput(Output<?> input); in addInput()
65 public OperationBuilder addInputList(Output<?>[] inputs); in addInputList()
80 public OperationBuilder addControlInput(Operation control); in addControlInput()
88 public OperationBuilder setDevice(String device); in setDevice()
97 public OperationBuilder setAttr(String name, String[] value); in setAttr()
106 public OperationBuilder setAttr(String name, String value); in setAttr()
115 public OperationBuilder setAttr(String name, byte[] value); in setAttr()
124 public OperationBuilder setAttr(String name, long value); in setAttr()
133 public OperationBuilder setAttr(String name, long[] value); in setAttr()
[all …]
DExecutionEnvironment.java30 OperationBuilder opBuilder(String type, String name); in opBuilder()
DEagerOperationBuilder.java24 final class EagerOperationBuilder implements OperationBuilder {
61 public OperationBuilder addControlInput(Operation control) { in addControlInput()
DEagerSession.java339 public OperationBuilder opBuilder(String type, String name) { in opBuilder()
DGraphOperationBuilder.java21 public final class GraphOperationBuilder implements OperationBuilder {
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/EDSC/
DIntrinsics.h18 using linalg_copy = OperationBuilder<linalg::CopyOp>;
19 using linalg_dot = OperationBuilder<linalg::DotOp>;
20 using linalg_fill = OperationBuilder<linalg::FillOp>;
21 using linalg_matmul = OperationBuilder<linalg::MatmulOp>;
22 using linalg_matvec = OperationBuilder<linalg::MatvecOp>;
23 using linalg_vecmat = OperationBuilder<linalg::VecmatOp>;
27 using linalg_yield = OperationBuilder<linalg::YieldOp>;
/external/llvm-project/mlir/include/mlir/Dialect/StandardOps/EDSC/
DIntrinsics.h21 using std_call = OperationBuilder<CallOp>;
27 using std_dealloc = OperationBuilder<DeallocOp>;
40 using std_ret = OperationBuilder<ReturnOp>;
45 using std_store = OperationBuilder<StoreOp>;
51 using std_tensor_store = OperationBuilder<TensorStoreOp>;
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/op/
DScope.java20 import org.tensorflow.OperationBuilder;
178 public OperationBuilder applyControlDependencies(OperationBuilder builder) { in applyControlDependencies()
DOperands.java21 import org.tensorflow.OperationBuilder;
/external/llvm-project/mlir/include/mlir/Dialect/Affine/EDSC/
DIntrinsics.h18 using affine_if = OperationBuilder<AffineIfOp>;
22 using affine_store = OperationBuilder<AffineStoreOp>;
/external/llvm-project/mlir/lib/Dialect/StandardOps/EDSC/
DIntrinsics.cpp16 return OperationBuilder<BranchOp>(block, operands); in std_br()
23 return OperationBuilder<CondBranchOp>(cond, trueBranch, trueOperands, in std_cond_br()
/external/llvm-project/mlir/include/mlir/Dialect/Vector/EDSC/
DIntrinsics.h31 using vector_print = OperationBuilder<vector::PrintOp>;
33 using vector_transfer_write = OperationBuilder<vector::TransferWriteOp>;
/external/llvm-project/mlir/include/mlir/Dialect/SCF/EDSC/
DIntrinsics.h18 using loop_yield = OperationBuilder<scf::YieldOp>;
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
DGraphOperationBuilderTest.java52 OperationBuilder b = in failOnUseAfterBuild()
65 OperationBuilder b = null; in failOnUseAfterGraphClose()
/external/llvm-project/mlir/lib/Conversion/LinalgToLLVM/
DLinalgToLLVM.cpp56 using llvm_call = OperationBuilder<LLVM::CallOp>;
59 using llvm_store = OperationBuilder<LLVM::StoreOp>;
67 using llvm_return = OperationBuilder<LLVM::ReturnOp>;
/external/llvm-project/mlir/docs/
DEDSC.md21 `mlir::edsc::OperationBuilder` can be used. Named intrinsics are provided as
26 using store = OperationBuilder<StoreOp>;
/external/llvm-project/mlir/include/mlir/EDSC/
DBuilders.h81 struct OperationBuilder { struct
83 OperationBuilder(Args... args) { in OperationBuilder() argument
/external/llvm-project/mlir/lib/Dialect/Linalg/Utils/
DUtils.cpp237 edsc::OperationBuilder<scf::ParallelOp>(
267 edsc::OperationBuilder<scf::ParallelOp>(