Home
last modified time | relevance | path

Searched defs:shape_invariant (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dset_shape_invariant_in_while_ops.cc34 auto shape_invariant = mlir::UnitAttr::get(&getContext()); in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID() local
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/
H A Dcontrol_flow.py210 name, check_shape, init, entry, exit_, shape_invariant): argument
1036 def _placeholder_value(like, shape_invariant, original=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_n_z.cc3058 bool shape_invariant) { in VerifyWhileTypes()