Searched refs:tuple_input (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 105 auto tuple_input = builder.create<mhlo::TupleOp>(loc, inputs); in LowerIf() local
110 tuple_input, tuple_input); in LowerIf()
130 auto tuple_input = builder.create<mhlo::TupleOp>(loc, inputs); in LowerCase() local
133 SmallVector<Value, 4> n_tuple_inputs(op.num_branches(), tuple_input); in LowerCase()
159 Value tuple_input = builder.create<mhlo::TupleOp>(loc, inputs); in LowerWhile() local
163 loc, builder.getTupleType(op.getResultTypes()), tuple_input); in LowerWhile()
222 Value tuple_input = builder->create<mhlo::TupleOp>(loc, implicit_inputs_ref); in TupleImplicitInputs() local
228 block.addArgument(tuple_input.getType()); in TupleImplicitInputs()
231 return tuple_input; in TupleImplicitInputs()
324 Value tuple_input = builder.create<mhlo::TupleOp>(loc, inputs); in LowerWhileRegion() local
[all …]
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | data_structures_test.py | 709 def f(tuple_input): argument
710 return tuple_input[0] + constant_op.constant(1.)
|