IterArgs (reference) in projects: external

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
D	lhlo_legalize_to_parallel_loops.cc	577 class IterArgs { class in mlir::lmhlo::__anonfcbbac650111::SelectAndScatterOpConverter 579 explicit IterArgs(ValueRange ivs_val_flag) : ivs_val_flag_(ivs_val_flag) {} in IterArgs() function in mlir::lmhlo::__anonfcbbac650111::SelectAndScatterOpConverter::IterArgs 580 IterArgs(ValueRange ivs, Value value, Value flag) { in IterArgs() function in mlir::lmhlo::__anonfcbbac650111::SelectAndScatterOpConverter::IterArgs 614 IterArgs ivs_val_flag(window_loops.inner_loop.getRegionIterArgs()); in SelectIvs() 642 IterArgs* ivs_val_flag, in SelectOrInitialize() 673 loc, IterArgs{operand_ivs, operand_elem, true_i1}.to_vector()); in SelectOrInitialize() 690 loc, IterArgs{operand_ivs, operand_elem, true_i1}.to_vector()); in SelectOrInitialize()