Home
last modified time | relevance | path

Searched refs:OpBuilder (Results 1 – 25 of 438) sorted by relevance

12345678910>>...18

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dop_factory.h24 class OpBuilder; variable
26 OpBuilder* CreateArgMinMaxOpBuilder(GraphBuilder* graph_builder, int op_type);
27 OpBuilder* CreateActivationBuilder(GraphBuilder* graph_builder, int op_type);
28 OpBuilder* CreateArithmeticBuilder(GraphBuilder* graph_builder, int op_type);
29 OpBuilder* CreateMatMulWithConstWeightsOpBuilder(GraphBuilder* graph_builder,
31 OpBuilder* CreateConcatBuilder(GraphBuilder* graph_builder, int op_type);
32 OpBuilder* CreateConv2DBuilder(GraphBuilder* graph_builder, int op_type);
33 OpBuilder* CreateTransposeConv2DBuilder(GraphBuilder* graph_builder,
35 OpBuilder* CreatePool2DBuilder(GraphBuilder* graph_builder, int op_type);
36 OpBuilder* CreateReshapeBuilder(GraphBuilder* graph_builder, int op_type);
[all …]
Dop_builder.cc75 OpBuilder* GraphBuilder::CreateOpBuilderFromTfLiteOp(int op_type, in CreateOpBuilderFromTfLiteOp()
172 OpBuilder* GraphBuilder::LookupConstData(uint64_t cache_key) { in LookupConstData()
178 void GraphBuilder::AddToCache(uint64_t cache_key, OpBuilder* value) { in AddToCache()
182 OpBuilder* GraphBuilder::AddConstNodeWithData(const int shape[], char* data, in AddConstNodeWithData()
186 builders_.emplace_back(new OpBuilder(this, OP_Const)); in AddConstNodeWithData()
201 OpBuilder* GraphBuilder::AddConstNodeWithData(int tensor_id, in AddConstNodeWithData()
219 builders_.emplace_back(new OpBuilder(this, OP_Const)); in AddConstNodeWithData()
235 OpBuilder* result_builder = builders_.back().get(); in AddConstNodeWithData()
251 OpBuilder** cast_op_builder) { in AddCastOp()
253 OpBuilder* cast_builder = CreateCastBuilder(this, op_type); in AddCastOp()
[all …]
Dconv_2d_builder.h32 OpBuilder* channel_scales_node = nullptr;
39 class Conv2dOpBuilder : public OpBuilder {
42 : OpBuilder(graph_builder, op_type) {} in Conv2dOpBuilder()
66 OpBuilder* weights_min_node_ = nullptr;
67 OpBuilder* weights_max_node_ = nullptr;
68 OpBuilder* bias_min_node_ = nullptr;
69 OpBuilder* bias_max_node_ = nullptr;
95 OpBuilder** bias_const_node = nullptr);
Dop_builder.h62 class OpBuilder {
67 OpBuilder(GraphBuilder* graph_builder, int hexagon_op_type) in OpBuilder() function
77 virtual ~OpBuilder() {} in ~OpBuilder()
235 OpBuilder* AddNodeFromTfLiteOp(int op_type, TfLiteNode* node,
241 OpBuilder* AddNode(int tflite_node_index = -1);
246 OpBuilder* AddConstNodeWithData(int tensor_id, const TfLiteTensor& tensor,
250 OpBuilder* AddConstNodeWithData(const int shape[], char* data, int data_size);
252 OpBuilder* CreateOpBuilderFromTfLiteOp(int op_type, TfLiteNode* node);
270 OpBuilder::TensorID GetHexagonTensorId(int tflite_tensor_index) { in GetHexagonTensorId()
273 return OpBuilder::TensorID(-1, -1); in GetHexagonTensorId()
[all …]
Dmatmul_builder.cc45 const OpBuilder::TensorID weights_id, in AddFullyConnectedHelper()
46 const OpBuilder::TensorID weights_min_id, in AddFullyConnectedHelper()
47 const OpBuilder::TensorID weights_max_id, in AddFullyConnectedHelper()
50 OpBuilder* matmul_op, in AddFullyConnectedHelper()
51 OpBuilder::TensorID* node_output) { in AddFullyConnectedHelper()
57 TF_LITE_ENSURE_STATUS(OpBuilder::ComputeMinAndMaxQuantValues( in AddFullyConnectedHelper()
67 matmul_op->AddInput(OpBuilder::TensorID(data_min_const->GetID(), 0)); in AddFullyConnectedHelper()
68 matmul_op->AddInput(OpBuilder::TensorID(data_max_const->GetID(), 0)); in AddFullyConnectedHelper()
89 OpBuilder::TensorID matmul_and_bias_out = matmul_out, in AddFullyConnectedHelper()
95 OpBuilder::ComputeMinAndMaxQuantValues(bias_tensor, &bias_min, &bias_max); in AddFullyConnectedHelper()
[all …]
Dmatmul_builder.h27 class MatMulWithConstWeightsOpBuilder : public OpBuilder {
31 : OpBuilder(graph_builder, op_type) {} in MatMulWithConstWeightsOpBuilder()
47 class MatMulOpBuilder : public OpBuilder {
50 : OpBuilder(graph_builder, op_type) {} in MatMulOpBuilder()
65 TfLiteContext* context, OpBuilder* matmul_op);
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dop_factory.h24 class OpBuilder; variable
26 OpBuilder* CreateAddOpBuilder(GraphBuilder* graph_builder);
27 OpBuilder* CreateAveragePool2dOpBuilder(GraphBuilder* graph_builder);
28 OpBuilder* CreateConcatenationOpBuilder(GraphBuilder* graph_builder);
29 OpBuilder* CreateConvolutionOpBuilder(GraphBuilder* graph_builder);
30 OpBuilder* CreateDepthwiseConvolutionOpBuilder(GraphBuilder* graph_builder);
31 OpBuilder* CreateFullyConnectedOpBuilder(GraphBuilder* graph_builder);
32 OpBuilder* CreateHardSwishOpBuilder(GraphBuilder* graph_builder);
33 OpBuilder* CreateLogisticOpBuilder(GraphBuilder* graph_builder);
34 OpBuilder* CreateMaxPool2dOpBuilder(GraphBuilder* graph_builder);
[all …]
Dop_builder.cc36 OpBuilder* GraphBuilder::AddBuilder(int builtin_code, const TfLiteNode* node) { in AddBuilder()
91 OpBuilder* GraphBuilder::AddBuilder( in AddBuilder()
92 const std::function<OpBuilder*(GraphBuilder*)>& builder, in AddBuilder()
98 OpBuilder* op = builder(this); in AddBuilder()
173 CoreML::Specification::NeuralNetworkLayer* OpBuilder::Build() { in Build()
178 TfLiteStatus OpBuilder::PopulateSubgraph(TfLiteContext* context) { in PopulateSubgraph()
183 void OpBuilder::SetBuiltinData(void* builtin_data) { in SetBuiltinData()
187 void OpBuilder::SetNodeID(int id) { node_id_ = id; } in SetNodeID()
189 void OpBuilder::SetTfLiteNode(const TfLiteNode* node) { tflite_node_ = node; } in SetTfLiteNode()
191 int OpBuilder::GetID() const { return node_id_; } in GetID()
[all …]
Dop_builder.h27 class OpBuilder; variable
59 OpBuilder* AddBuilder(int builtin_code, const TfLiteNode* node);
62 OpBuilder* AddBuilder(const std::function<OpBuilder*(GraphBuilder*)>& builder,
87 std::vector<std::unique_ptr<OpBuilder>> builders_;
96 class OpBuilder {
98 explicit OpBuilder(GraphBuilder* graph_builder) in OpBuilder() function
100 virtual ~OpBuilder() {} in ~OpBuilder()
Dactivation_layer_builder.cc114 OpBuilder* CreateActivationLayerBuilder(GraphBuilder* graph_builder) { in CreateActivationLayerBuilder()
118 OpBuilder* CreateLogisticOpBuilder(GraphBuilder* graph_builder) { in CreateLogisticOpBuilder()
122 OpBuilder* CreateReluOpBuilder(GraphBuilder* graph_builder) { in CreateReluOpBuilder()
126 OpBuilder* CreateReluN1To1OpBuilder(GraphBuilder* graph_builder) { in CreateReluN1To1OpBuilder()
130 OpBuilder* CreateRelu6OpBuilder(GraphBuilder* graph_builder) { in CreateRelu6OpBuilder()
134 OpBuilder* CreateTanhOpBuilder(GraphBuilder* graph_builder) { in CreateTanhOpBuilder()
/external/llvm-project/mlir/include/mlir/Conversion/StandardToLLVM/
DConvertStandardToLLVM.h111 OpBuilder &builder);
117 OpBuilder &builder);
223 static StructBuilder undef(OpBuilder &builder, Location loc,
236 Value extractPtr(OpBuilder &builder, Location loc, unsigned pos);
238 void setPtr(OpBuilder &builder, Location loc, unsigned pos, Value ptr);
246 static ComplexStructBuilder undef(OpBuilder &builder, Location loc,
250 Value real(OpBuilder &builder, Location loc);
252 void setReal(OpBuilder &builder, Location loc, Value real);
255 Value imaginary(OpBuilder &builder, Location loc);
257 void setImaginary(OpBuilder &builder, Location loc, Value imaginary);
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcollection_ops_util.h36 Value CreateScalarConst(int32_t value, OpBuilder builder, Location loc);
39 Value GetR1Const(ArrayRef<int64_t> r1, OpBuilder builder, Location loc,
45 TensorType GetSizeType(OpBuilder builder);
48 Value ReshapeScalarToSizeType(OpBuilder builder, Value scalar, Location loc);
52 Value GetIndicesForElement(Value index, Value buffer, OpBuilder builder,
57 Value GetElement(Value index, Value buffer, OpBuilder builder, Location loc,
62 Value SetElement(Value index, Value buffer, Value element, OpBuilder builder,
69 Type element_dtype, OpBuilder builder,
75 Type element_dtype, OpBuilder builder,
86 Value ReadLocalVariable(Value local_var, OpBuilder builder, Location loc);
[all …]
/external/llvm-project/mlir/lib/EDSC/
DBuilders.cpp18 mlir::edsc::ScopedContext::ScopedContext(OpBuilder &b) in ScopedContext()
21 mlir::edsc::ScopedContext::ScopedContext(OpBuilder &b, Location location) in ScopedContext()
30 mlir::edsc::ScopedContext::ScopedContext(OpBuilder &b, in ScopedContext()
31 OpBuilder::InsertPoint newInsertPt, in ScopedContext()
48 OpBuilder &mlir::edsc::ScopedContext::getBuilderRef() { in getBuilderRef()
66 OpBuilder &builder = ScopedContext::getBuilderRef(); in createBlock()
76 OpBuilder &builder = ScopedContext::getBuilderRef(); in createBlockInRegion()
78 OpBuilder::InsertionGuard guard(builder); in createBlockInRegion()
85 OpBuilder &builder = ScopedContext::getBuilderRef(); in appendToBlock()
87 OpBuilder::InsertionGuard guard(builder); in appendToBlock()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/python/mlir_wrapper/
Dbuilders.cc31 py::class_<mlir::OpBuilder>(m, "OpBuilder") in init_builders()
36 .def("getUnknownLoc", &mlir::OpBuilder::getUnknownLoc) in init_builders()
39 &mlir::OpBuilder::setInsertionPoint)) in init_builders()
40 .def("saveInsertionPoint", &mlir::OpBuilder::saveInsertionPoint) in init_builders()
41 .def("restoreInsertionPoint", &mlir::OpBuilder::restoreInsertionPoint) in init_builders()
44 [](mlir::OpBuilder& opb, mlir::OperationState& state) { in init_builders()
48 .def("getContext", &mlir::OpBuilder::getContext, in init_builders()
51 py::class_<mlir::OpBuilder::InsertPoint>(m, "OpBuilder_InsertionPoint") in init_builders()
52 .def("getBlock", &mlir::OpBuilder::InsertPoint::getBlock); in init_builders()
Dops.cc79 [](mlir::OpBuilder& opb, mlir::Location loc, in init_ops()
90 [](mlir::OpBuilder& opb, mlir::Location loc, mlir::Value x, in init_ops()
97 [](mlir::OpBuilder& opb, mlir::Location loc, mlir::Value input, in init_ops()
109 [](mlir::OpBuilder& opb, mlir::Location loc, in init_ops()
117 [](mlir::OpBuilder& opb, mlir::Location loc, mlir::Value x, in init_ops()
127 [](mlir::OpBuilder& opb, mlir::Location loc, mlir::Value x, in init_ops()
136 [](mlir::OpBuilder& opb, mlir::Location loc, mlir::Value x, in init_ops()
144 [](mlir::OpBuilder& opb, mlir::Location loc, in init_ops()
157 [](mlir::OpBuilder& opb, mlir::Location loc, mlir::Value x, in init_ops()
165 [](mlir::OpBuilder& opb, mlir::Location loc, mlir::Value x, in init_ops()
[all …]
/external/llvm-project/mlir/include/mlir/Dialect/Tosa/Utils/
DQuantUtils.h35 ConvOpQuantizationAttr buildConvOpQuantizationAttr(OpBuilder &builder,
39 MatMulOpQuantizationAttr buildMatMulOpQuantizationAttr(OpBuilder &builder,
43 UnaryOpQuantizationAttr buildUnaryOpQuantizationAttr(OpBuilder &builder,
48 PadOpQuantizationAttr buildPadOpQuantizationAttr(OpBuilder &builder,
53 Type buildConvOpResultTypeInfo(OpBuilder &builder, Type outputType, Value input,
57 Type buildQTypeFromMinMax(OpBuilder builder, Type inputDType, Attribute minAttr,
63 TypeAttr buildQTypeAttrFromMinMax(OpBuilder builder, Type inputDType,
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/transforms/
Dmap_lmhlo_to_scalar_op.h89 ArrayRef<Value> args, OpBuilder* b) {
97 ArrayRef<Value> args, OpBuilder* b) {
105 ArrayRef<Value> args, OpBuilder* b) {
119 ArrayRef<Value> args, OpBuilder* b) {
129 OpBuilder* b) {
156 OpBuilder* b) {
167 OpBuilder* b) {
176 OpBuilder* b) {
216 ArrayRef<Value> args, OpBuilder* b) {
241 OpBuilder* b) {
[all …]
/external/llvm-project/mlir/include/mlir/IR/
DBuilders.h179 class OpBuilder : public Builder {
184 explicit OpBuilder(MLIRContext *ctx, Listener *listener = nullptr)
188 explicit OpBuilder(Region *region, Listener *listener = nullptr)
189 : OpBuilder(region->getContext(), listener) {
193 explicit OpBuilder(Region &region, Listener *listener = nullptr)
194 : OpBuilder(&region, listener) {}
198 explicit OpBuilder(Operation *op, Listener *listener = nullptr)
199 : OpBuilder(op->getContext(), listener) {
203 OpBuilder(Block *block, Block::iterator insertPoint,
205 : OpBuilder(block->getParent()->getContext(), listener) {
[all …]
/external/llvm-project/mlir/test/mlir-tblgen/
Dop-decl.td38 OpBuilder<"double deprecatedForm">];
85 // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, Valu…
86 // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, int …
87 // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, doub…
88 // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::ml…
89 // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::ml…
90 // CHECK: static void build(::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRa…
156 // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, /*op…
186 // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir…
187 // CHECK: static void build(::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir…
[all …]
/external/llvm-project/mlir/lib/Dialect/SCF/EDSC/
DBuilders.cpp25 steps, [&](OpBuilder &builder, Location loc, ValueRange ivs) { in loopNestBuilder()
54 [&](OpBuilder &builder, Location loc, ValueRange ivs, ValueRange args) { in loopNestBuilder()
74 [&](OpBuilder &builder, Location loc, ValueRange ivs, ValueRange args) { in loopNestBuilder()
83 static std::function<void(OpBuilder &, Location)>
86 return [=](OpBuilder &builder, Location loc) { in wrapIfBody()
111 static std::function<void(OpBuilder &, Location)>
113 return [=](OpBuilder &builder, Location loc) { in wrapZeroResultIfBody()
129 elseBody ? llvm::function_ref<void(OpBuilder &, Location)>( in conditionBuilder()
131 : llvm::function_ref<void(OpBuilder &, Location)>(nullptr)); in conditionBuilder()
/external/llvm-project/mlir/include/mlir/Conversion/AffineToStandard/
DAffineToStandard.h22 class OpBuilder; variable
33 Value expandAffineExpr(OpBuilder &builder, Location loc, AffineExpr expr,
38 Optional<SmallVector<Value, 8>> expandAffineMap(OpBuilder &builder,
56 Value lowerAffineLowerBound(AffineForOp op, OpBuilder &builder);
60 Value lowerAffineUpperBound(AffineForOp op, OpBuilder &builder);
/external/llvm-project/mlir/include/mlir/Dialect/SCF/
DSCF.h26 void buildTerminatedBody(OpBuilder &builder, Location loc);
79 OpBuilder &builder, Location loc, ValueRange lbs, ValueRange ubs,
81 function_ref<ValueVector(OpBuilder &, Location, ValueRange, ValueRange)>
88 LoopNest buildLoopNest(OpBuilder &builder, Location loc, ValueRange lbs,
90 function_ref<void(OpBuilder &, Location, ValueRange)>
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dchlo_ops.h57 static Value getConstantLike(OpBuilder& b, Location loc, T constant, in getConstantLike()
69 Value getConstantLike(OpBuilder& b, Location loc, const APFloat& constant,
72 Value getConstantLikeMaxFiniteValue(OpBuilder& b, Location loc, Value val);
74 Value getConstantLikeInfValue(OpBuilder& b, Location loc, Value val,
77 Value getConstantLikeSmallestFiniteValue(OpBuilder& b, Location loc, Value val);
/external/llvm-project/mlir/include/mlir/Dialect/StandardOps/IR/
DOps.h34 class OpBuilder; variable
42 OpBuilder &b, Location loc);
59 static void build(OpBuilder &builder, OperationState &result,
78 static void build(OpBuilder &builder, OperationState &result, int64_t value,
83 static void build(OpBuilder &builder, OperationState &result, int64_t value,
103 static void build(OpBuilder &builder, OperationState &result, int64_t value);
154 static void build(OpBuilder &builder, OperationState &result, Value srcMemRef,
277 static void build(OpBuilder &builder, OperationState &result, Value tagMemRef,
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_legalize_to_parallel_loops.cc43 Block* lhlo_block, OpBuilder* b) { in ApplySingleResultLhloCode()
67 Block* lhlo_block, OpBuilder* b) { in ConvertToReductionOperator()
69 OpBuilder::InsertionGuard guard(*b); in ConvertToReductionOperator()
79 size_t dim_index, int64_t dim, OpBuilder* b) { in GetStaticOrDynamicDim()
94 OpBuilder* b) { in MapWindowIvsToInput()
140 OpBuilder* b) { in MakeLoopOverShape()
440 OpBuilder then_builder = in CreateReduceOpInNestedParallelLoops()
446 OpBuilder else_builder = in CreateReduceOpInNestedParallelLoops()
506 OpBuilder rmw_builder = OpBuilder::atBlockEnd(rmw.getBody()); in matchAndRewrite()
518 OpBuilder* b) const { in InitializeOutput()
[all …]

12345678910>>...18