Home
last modified time | relevance | path

Searched refs:yield_op (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Drecord_input_test.py51 yield_op = data_flow_ops.RecordInput(
58 self.assertEqual(self.evaluate(yield_op), b"0000000000")
68 yield_op = data_flow_ops.RecordInput(
77 self.assertEqual(self.evaluate(yield_op), b"0000000000")
87 yield_op = data_flow_ops.RecordInput(
96 self.assertEqual(self.evaluate(yield_op), b"0000000000")
116 yield_op = records.get_yield_op()
122 op_list = self.evaluate(yield_op)
142 yield_op = records.get_yield_op()
144 self.evaluate(yield_op)
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dwhile_loop_outline.cc184 Operation* yield_op = outlined_func.getBody().front().getTerminator(); in OutlineWhile() local
185 OpBuilder b(yield_op); in OutlineWhile()
188 yield_op->getOperands().take_front(num_loop_carried); in OutlineWhile()
200 auto cast = b.create<CastOp>(yield_op->getLoc(), type, value); in OutlineWhile()
203 auto cast = b.create<TF::CastOp>(yield_op->getLoc(), type, value); in OutlineWhile()
210 args.append(yield_op->operand_begin(), yield_op->operand_end()); in OutlineWhile()
212 b.create<ReturnOp>(yield_op->getLoc(), args); in OutlineWhile()
213 yield_op->erase(); in OutlineWhile()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dexecutor_tpuv1_outline_tpu_island.cc116 YieldOp yield_op = island_op.GetYield(); in runOnOperation() local
120 OpBuilder replacer(yield_op); in runOnOperation()
122 replacer.create<ReturnOp>(yield_op.getLoc(), yield_op.getOperands()); in runOnOperation()
123 yield_op.erase(); in runOnOperation()
Dexecutor_tpuv1_island_coarsening.cc237 YieldOp yield_op = island.GetYield(); in MergeIsland() local
244 use.set(yield_op.getOperand(idx_result.index())); in MergeIsland()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_executor.cc326 YieldOp yield_op = op.GetYield(); in Print() local
331 yield_op.getLoc() == op.getLoc()) { in Print()
987 YieldOp yield_op = island_op.GetYield(); in matchAndRewrite() local
1001 new_rets.push_back(yield_op.getOperand(result.getResultNumber())); in matchAndRewrite()