Searched defs:shape_invariant (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | set_shape_invariant_in_while_ops.cc | 30 auto shape_invariant = mlir::UnitAttr::get(&getContext()); in runOnOperation() local
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow.py | 212 name, check_shape, init, entry, exit_, shape_invariant): argument
|
| D | control_flow_deprecated_py2.py | 137 name, check_shape, init, entry, exit_, shape_invariant): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 2904 bool shape_invariant) { in VerifyWhileTypes()
|