| /external/llvm-project/mlir/include/mlir/Conversion/StandardToLLVM/ |
| D | ConvertStandardToLLVM.h | 565 template <typename SourceOp>
570 : ConvertToLLVMPattern(SourceOp::getOperationName(),
577 rewrite(cast<SourceOp>(op), operands, rewriter); in rewrite()
580 return match(cast<SourceOp>(op)); in match()
585 return matchAndRewrite(cast<SourceOp>(op), operands, rewriter); in matchAndRewrite()
590 virtual void rewrite(SourceOp op, ArrayRef<Value> operands, in rewrite()
594 virtual LogicalResult match(SourceOp op) const { in match()
598 matchAndRewrite(SourceOp op, ArrayRef<Value> operands, in matchAndRewrite()
633 template <typename SourceOp, typename TargetOp>
634 class OneToOneConvertToLLVMPattern : public ConvertOpToLLVMPattern<SourceOp> {
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer_test.cc | 409 Node* input = ops::SourceOp("TestInput", b.opts().WithName("in")); in TEST_F()
428 Node* input = ops::SourceOp("TestInput", b.opts().WithName("in")); in TEST_F()
448 Node* input = ops::SourceOp("TestInput", b.opts().WithName("in")); in TEST_F()
472 Node* input = ops::SourceOp("TestGPUOutput", b.opts().WithName("in")); in TEST_F()
489 Node* input = ops::SourceOp("TestGPUOutput", b.opts().WithName("in")); in TEST_F()
551 Node* ds = ops::SourceOp("CreateDatasetSP", b.opts().WithName("ds")); in TEST_F()
566 Node* ds = ops::SourceOp("CreateDatasetRP", b.opts().WithName("ds")); in TEST_F()
581 Node* ds = ops::SourceOp("CreateDatasetNP", b.opts().WithName("ds")); in TEST_F()
596 Node* ds = ops::SourceOp("CreateDatasetNP", b.opts().WithName("ds")); in TEST_F()
611 Node* ds = ops::SourceOp("CreateDatasetSP", b.opts().WithName("ds")); in TEST_F()
[all …]
|
| /external/llvm-project/mlir/lib/Conversion/GPUToSPIRV/ |
| D | ConvertGPUToSPIRV.cpp | 27 template <typename SourceOp, spirv::BuiltIn builtin>
28 class LaunchConfigConversion : public SPIRVOpLowering<SourceOp> {
30 using SPIRVOpLowering<SourceOp>::SPIRVOpLowering;
33 matchAndRewrite(SourceOp op, ArrayRef<Value> operands,
39 template <typename SourceOp, spirv::BuiltIn builtin>
40 class SingleDimLaunchConfigConversion : public SPIRVOpLowering<SourceOp> {
42 using SPIRVOpLowering<SourceOp>::SPIRVOpLowering;
45 matchAndRewrite(SourceOp op, ArrayRef<Value> operands,
117 template <typename SourceOp, spirv::BuiltIn builtin>
118 LogicalResult LaunchConfigConversion<SourceOp, builtin>::matchAndRewrite( in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_quantize_helper.h | 133 template <typename SourceOp>
134 struct PrepareLstmOutputScale : public OpRewritePattern<SourceOp> {
137 : OpRewritePattern<SourceOp>(context) {} in PrepareLstmOutputScale()
138 LogicalResult matchAndRewrite(SourceOp op, in matchAndRewrite()
170 LogicalResult processRestrictScale(SourceOp op, int input_index, in processRestrictScale()
222 template <typename SourceOp>
223 struct ConvertOpStatsToQDQs : public OpRewritePattern<SourceOp> {
228 : OpRewritePattern<SourceOp>(context, benefit),
235 SourceOp op, const operator_property::OpVariant& op_variant, in processInputs()
280 SourceOp op, Operation* const_op, int input_index, in processConstantOp()
[all …]
|
| D | legalize_tf.cc | 581 template <typename SourceOp>
582 class ApplyExplicitBroadcasting : public OpRewritePattern<SourceOp> {
584 using OpRewritePattern<SourceOp>::OpRewritePattern;
586 LogicalResult matchAndRewrite(SourceOp src_op, in matchAndRewrite()
639 rewriter.replaceOpWithNewOp<SourceOp>(op, result_type, lhs, rhs); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | OpToFuncCallLowering.h | 31 template <typename SourceOp>
36 : ConvertToLLVMPattern(SourceOp::getOperationName(), in OpToFuncCallLowering()
47 std::is_base_of<OpTrait::OneResult<SourceOp>, SourceOp>::value, in matchAndRewrite()
50 static_assert(std::is_base_of<OpTrait::SameOperandsAndResultType<SourceOp>, in matchAndRewrite()
51 SourceOp>::value, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | cluster_scoping_pass_test.cc | 95 Node* a = ops::SourceOp("Const", builder.opts() in TEST()
99 Node* b = ops::SourceOp("Const", builder.opts() in TEST()
103 Node* unstage = ops::SourceOp( in TEST()
139 Node* a = ops::SourceOp("Const", builder.opts() in TEST()
143 Node* b = ops::SourceOp("Const", builder.opts() in TEST()
147 Node* unstage = ops::SourceOp( in TEST()
|
| D | partially_decluster_pass_test.cc | 129 ops::SourceOp("FakeNullary", builder.opts().WithName("Input")); in TEST()
165 ops::SourceOp("FakeNullary", builder.opts().WithName("Input")); in TEST()
194 ops::SourceOp("FakeNullary", builder.opts().WithName("Input")); in TEST()
224 ops::SourceOp("FakeNullary", builder.opts().WithName("Input")); in TEST()
225 Node* resource_var = ops::SourceOp("FakeResourceVar", in TEST()
255 ops::SourceOp("FakeNullary", builder.opts().WithName("Input")); in TEST()
417 ops::SourceOp("FakeNullary", builder.opts().WithName("Input")); in TEST()
|
| D | mark_for_compilation_pass_test.cc | 98 ops::SourceOp("UncompilableNullary", builder.opts().WithName("A")); in TEST()
122 Node* a = ops::SourceOp("Const", builder.opts() in TEST()
142 Node* a = ops::SourceOp("Const", builder.opts() in TEST()
163 Node* a = ops::SourceOp( in TEST()
184 Node* a = ops::SourceOp("Const", builder.opts() in TEST()
258 ops::SourceOp("UncompilableNullary", builder.opts().WithName("A")); in TEST()
295 ops::SourceOp("VarHandleOp", builder.opts() in TEST()
302 Node* value = ops::SourceOp("Const", builder.opts() in TEST()
384 ops::SourceOp("UncompilableNullary", builder.opts().WithName("A")); in TEST()
450 Node* a = ops::SourceOp("Const", builder.opts() in TEST()
[all …]
|
| D | compilability_check_util_test.cc | 119 Node* const0 = ops::SourceOp("InputFloatOp", opts); in TEST_F()
162 Node* const0 = ops::SourceOp("InputFloatOp", builder.opts()); in TEST_F()
213 Node* const0 = ops::SourceOp("InputFloatOp", builder.opts()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | collective_order_test.cc | 114 Node* a = ops::SourceOp("TestParams", in InitGraph()
116 Node* b = ops::SourceOp("TestParams", in InitGraph()
169 Node* a = ops::SourceOp("TestParams", in InitGraph2()
205 Node* w = ops::SourceOp("TestParams", in InitGraphForPruning()
207 Node* x = ops::SourceOp("TestParams", in InitGraphForPruning()
209 Node* y = ops::SourceOp("TestParams", in InitGraphForPruning()
211 Node* z = ops::SourceOp("TestParams", in InitGraphForPruning()
|
| D | algorithm_test.cc | 83 Node* w1 = SourceOp("TestParams", b.opts().WithName("W1")); in TEST()
84 Node* w2 = SourceOp("TestParams", b.opts().WithName("W2")); in TEST()
86 SourceOp("TestInput", b.opts().WithName("input").WithControlInput(w1)); in TEST()
135 Node* w1 = SourceOp("TestParams", b.opts().WithName("W1")); in TEST()
137 SourceOp("TestInput", b.opts().WithName("input").WithControlInput(w1)); in TEST()
162 Node* n0 = ops::SourceOp("TestParams", b.opts().WithName("n0")); in TEST()
|
| D | graph_def_builder.h | 193 Node* SourceOp(const string& op_name, const GraphDefBuilder::Options& opts);
|
| D | graph_def_builder.cc | 101 Node* SourceOp(const string& op_name, const GraphDefBuilder::Options& opts) { in SourceOp() function
|
| D | subgraph_test.cc | 362 last_node = ops::SourceOp("In", b.opts().WithName(name)); in BM_SubgraphHelper()
|
| /external/llvm-project/mlir/include/mlir/Dialect/SPIRV/ |
| D | SPIRVLowering.h | 67 template <typename SourceOp>
68 class SPIRVOpLowering : public OpConversionPattern<SourceOp> {
72 : OpConversionPattern<SourceOp>(context, benefit),
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | PatternMatch.h | 193 template <typename SourceOp>
198 : RewritePattern(SourceOp::getOperationName(), benefit, context) {} in RewritePattern()
202 rewrite(cast<SourceOp>(op), rewriter); in rewrite()
205 return match(cast<SourceOp>(op)); in match()
209 return matchAndRewrite(cast<SourceOp>(op), rewriter); in matchAndRewrite()
214 virtual void rewrite(SourceOp op, PatternRewriter &rewriter) const { in rewrite()
217 virtual LogicalResult match(SourceOp op) const { in match()
220 virtual LogicalResult matchAndRewrite(SourceOp op, in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/ |
| D | DecomposeCallGraphTypes.cpp | 35 template <typename SourceOp>
37 : public OpConversionPattern<SourceOp> {
43 : OpConversionPattern<SourceOp>(typeConverter, context, benefit), in DecomposeCallGraphTypesOpConversionPattern()
|
| /external/llvm/utils/TableGen/ |
| D | PseudoLoweringEmitter.cpp | 179 StringMap<unsigned>::iterator SourceOp = in evaluateExpansion() local
181 if (SourceOp == SourceOperands.end()) in evaluateExpansion()
189 SourceOp->getValue(); in evaluateExpansion()
191 DEBUG(dbgs() << " " << SourceOp->getValue() << " ==> " << i << "\n"); in evaluateExpansion()
|
| /external/llvm-project/llvm/utils/TableGen/ |
| D | PseudoLoweringEmitter.cpp | 197 StringMap<unsigned>::iterator SourceOp = in evaluateExpansion() local
199 if (SourceOp == SourceOperands.end()) { in evaluateExpansion()
210 SourceOp->getValue(); in evaluateExpansion()
212 LLVM_DEBUG(dbgs() << " " << SourceOp->getValue() << " ==> " << i in evaluateExpansion()
|
| D | RISCVCompressInstEmitter.cpp | 376 StringMap<unsigned>::iterator SourceOp = in createInstOperandMapping() local
378 if (SourceOp == SourceOperands.end()) in createInstOperandMapping()
385 SourceDag->getArgNameStr(SourceOp->getValue()) && in createInstOperandMapping()
387 DestOperandMap[i].Data.Operand = SourceOp->getValue(); in createInstOperandMapping()
388 SourceOperandMap[SourceOp->getValue()].Data.Operand = i; in createInstOperandMapping()
389 LLVM_DEBUG(dbgs() << " " << SourceOp->getValue() << " ====> " << i in createInstOperandMapping()
|
| /external/llvm-project/mlir/include/mlir/Transforms/ |
| D | DialectConversion.h | 374 template <typename SourceOp>
377 : ConversionPattern(SourceOp::getOperationName(), benefit, context) {} in ConversionPattern()
380 : ConversionPattern(SourceOp::getOperationName(), benefit, typeConverter, in ConversionPattern()
387 rewrite(cast<SourceOp>(op), operands, rewriter); in rewrite()
392 return matchAndRewrite(cast<SourceOp>(op), operands, rewriter); in matchAndRewrite()
399 virtual void rewrite(SourceOp op, ArrayRef<Value> operands, in rewrite()
405 matchAndRewrite(SourceOp op, ArrayRef<Value> operands, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | equal_graph_def_test.cc | 36 return ops::SourceOp("Input", opts); in Input()
40 return ops::SourceOp("Alternate", opts); in Alternate()
|
| /external/llvm-project/mlir/lib/Conversion/AVX512ToLLVM/ |
| D | ConvertAVX512ToLLVM.cpp | 35 template <typename SourceOp, typename TargetOp>
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 317 *output = ops::SourceOp( in AddPlaceholderInternal()
324 *output = ops::SourceOp( in AddTensorInternal()
|