Home
last modified time | relevance | path

Searched refs:getOperation (Results 1 – 25 of 92) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlhlo_fuse_linalg.cc55 auto op = cast<LinalgOp>(generic_op.getOperation()); in runOnFunction()
75 auto originalOp = info->originalProducer.getOperation(); in runOnFunction()
80 *originalOpInLinalgOpsVector = info->fusedProducer.getOperation(); in runOnFunction()
Dlegalize_control_flow.cc64 Operation* op_inst = conditional_op.getOperation(); in LowerConditionalOp()
117 auto* op_inst = while_op.getOperation(); in LowerWhileOp()
Dxla_legalize_to_linalg.cc59 op.getOperation()->getOperand(0).getType().template cast<ShapedType>(); in matchAndRewrite()
101 Value result = op.getOperation()->getResult(0); in matchAndRewrite()
136 rewriter.replaceOp(op, linalgOp.getOperation()->getResults()); in matchAndRewrite()
Dunfuse_batch_norm.cc100 auto epsilon = MaterializeEpsilon(bn_op.getOperation(), bn_op.epsilonAttr(), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_ophint_func_op.cc113 builder->setInsertionPoint(call_op.getOperation()); in BuildUnidirectionalSequenceRnnOp()
145 builder->setInsertionPoint(call_op.getOperation()); in BuildUnidirectionalSequenceLSTMOp()
186 builder->setInsertionPoint(call_op.getOperation()); in BuildUnidirectionalSequenceLSTMOp()
219 Operation* call = call_op.getOperation(); in ConvertTfLiteFusedOpIfAvailable()
252 Operation* func = composite_func_op.getOperation(); in ConvertCallOps()
Ddilated_conv.h85 Operation* prev_op = op.getOperation()->getPrevNode(); in matchAndRewrite()
88 Operation* next_op = op.getOperation()->getNextNode(); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc225 .getOperation(); in ImportInstruction()
312 .getOperation(); in ImportInstruction()
346 .getOperation(); in ImportInstruction()
361 .getOperation(); in ImportInstruction()
374 return op.getOperation(); in ImportInstruction()
383 .getOperation(); in ImportInstruction()
395 return reduce.getOperation(); in ImportInstruction()
402 .getOperation(); in ImportInstruction()
412 .getOperation(); in ImportInstruction()
418 .getOperation(); in ImportInstruction()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dsink_constant.cc45 LLVM_DEBUG(llvm::dbgs() << "Visit " << *launch.getOperation() << "\n"); in runOnFunction()
71 const_op.getOperation()->moveBefore(&body.begin()->front()); in runOnFunction()
Dfunctional_control_flow_to_cfg.cc89 return builder->create<CallOp>(loc, fn, operands).getOperation(); in CallFn()
156 Operation* op_inst = op.getOperation(); in LowerIfOp()
217 Operation* op_inst = op.getOperation(); in LowerWhileOp()
Dgraph_pruning.cc68 Operation* sink_op = source_op.GetSink().getOperation(); in PruneGraph()
Dreplicate_invariant_op_hoisting.cc106 shape_op.replaceAllUsesWith(new_shape_op.getOperation()); in MakeShapeOpInvariant()
Dexecutor_island_coarsening.cc249 MoveInnerOpsToNewIsland(parent, child, new_yield_op.getOperation()); in MergeIslands()
315 fetch.getOperation()->eraseOperand(fetch.getNumOperands() - 1); in InsertDummyIslandForFetch()
Dtpu_rewrite_pass.cc111 module_for_func.get().getOperation()->setAttr(kVersionsAttr, versions_attr); in EncapsulateFuncAndSerialize()
457 Block* block = launch_func.getOperation()->getBlock(); in Rewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dbreakup-islands.cc55 auto graph_op_range = getOperation().getBody().front().without_terminator(); in runOnOperation()
60 getOperation().getBody().front().front()); in runOnOperation()
63 getOperation().emitError("Expected function to contain only a graph_op"); in runOnOperation()
80 OpBuilder builder(getOperation()); in runOnOperation()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dkernel_lowering.cc150 auto* block = loadOp.getOperation()->getBlock(); in runOnFunction()
153 return &other == loadOp.getOperation(); in runOnFunction()
188 return allocOp.getOperation(); in SearchAllocOp()
324 ::mlir::gpu::GPUModuleOp m = getOperation(); in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_device.cc90 const auto& regions = op.getOperation()->getRegions(); in Verify()
117 op.getOperation()->getResult(output_index++).getType()) { in Verify()
126 const int num_output_types = op.getOperation()->getNumResults(); in Verify()
160 return getOperation()->getRegion(index).front(); in getRegionWithIndex()
Dtf_executor.cc241 p.printRegion(graph.getOperation()->getRegion(0)); in Print()
389 p.printRegion(op.getOperation()->getRegion(0)); in Print()
522 p.printFunctionalType(switch_op.getOperation()); in Print()
696 p.printFunctionalType(merge.getOperation()); in Print()
764 p.printFunctionalType(enter.getOperation()); in Print()
995 p.printFunctionalType(loop_cond.getOperation()); in Print()
1117 Operation *operation = op.getOperation(); in matchAndRewrite()
Dtf_device_ops.td67 Block &GetBody() { return getOperation()->getRegion(0).front(); }
182 Block &GetBody() { return getOperation()->getRegion(0).front(); }
/external/llvm/lib/Target/AArch64/MCTargetDesc/
DAArch64AsmBackend.cpp407 switch (Inst.getOperation()) { in generateCompactUnwindEncoding()
419 assert(LRPush.getOperation() == MCCFIInstruction::OpOffset && in generateCompactUnwindEncoding()
422 assert(FPPush.getOperation() == MCCFIInstruction::OpOffset && in generateCompactUnwindEncoding()
452 if (Inst2.getOperation() != MCCFIInstruction::OpOffset) in generateCompactUnwindEncoding()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc350 [](APInt a, APInt b) { return a + b; }, getOperation()->isCommutative()); in fold()
468 return VerifyConcatenationOpTypes(op.getOperation(), output_type, in Verify()
549 mlir::OpBuilder builder(getOperation()); in fold()
678 [](APInt a, APInt b) { return a * b; }, getOperation()->isCommutative()); in fold()
691 getOperation()->isCommutative()); in fold()
704 if (op.getOperation()->getNumOperands() != op.values_count()) in Verify()
979 [](APInt a, APInt b) { return a - b; }, getOperation()->isCommutative()); in fold()
1065 if (op.getOperation()->getNumResults() != op.num()) in Verify()
1144 op.getOperation(), num_splits, in Verify()
1231 return VerifySplitOpOutputTypes(op.getOperation(), num_splits, in Verify()
[all …]
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Target/AArch64/MCTargetDesc/
DAArch64AsmBackend.cpp456 switch (Inst.getOperation()) { in generateCompactUnwindEncoding()
476 assert(LRPush.getOperation() == MCCFIInstruction::OpOffset && in generateCompactUnwindEncoding()
479 assert(FPPush.getOperation() == MCCFIInstruction::OpOffset && in generateCompactUnwindEncoding()
509 if (Inst2.getOperation() != MCCFIInstruction::OpOffset) in generateCompactUnwindEncoding()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddump_mlir_util_test.cc57 module_ref->getOperation()->print(os); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_traits.h71 auto op = this->getOperation(); in GetResultQuantizedType()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tensorflow/
Dtf_to_quant.cc128 quantize.getOperation()->replaceUsesOfWith(dequantize, value); in matchAndRewrite()
/external/mesa3d/src/gallium/drivers/nouveau/codegen/
Dnv50_ir_from_nir.cpp143 operation getOperation(nir_intrinsic_op);
144 operation getOperation(nir_op);
145 operation getOperation(nir_texop);
378 Converter::getOperation(nir_op op) in getOperation() function in __anon87abe68c0111::Converter
500 Converter::getOperation(nir_texop op) in getOperation() function in __anon87abe68c0111::Converter
530 Converter::getOperation(nir_intrinsic_op op) in getOperation() function in __anon87abe68c0111::Converter
1974 mkOp1(getOperation(op), TYPE_U32, NULL, mkImm(idx))->fixed = 1; in visit()
1981 mkOp1(getOperation(op), TYPE_U32, NULL, mkImm(idx))->fixed = 1; in visit()
2264 TexInstruction *texi = mkTex(getOperation(op), target.getEnum(), location, 0, defs, srcs); in visit()
2519 Instruction *i1 = mkOp(getOperation(op), dType, newDefs[0]); in visit()
[all …]

1234