Searched refs:scalar_pred (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | stable_sort_expander.cc | 181 Shape scalar_pred = ShapeUtil::MakeShape(PRED, {}); in ExpandInstruction() local
184 scalar_pred, old_root, cloned_root, ComparisonDirection::kEq)); in ExpandInstruction()
187 scalar_pred, comparator->parameter_instruction(2 * iota_index), in ExpandInstruction()
|
| D | while_util.cc | 156 Shape scalar_pred = ShapeUtil::MakeShape(PRED, {}); in MakeCountedLoopConditionComputation() local
160 {&loop_state_shape}, scalar_pred, "while_cond")); in MakeCountedLoopConditionComputation()
|
| /external/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_nir_soa.c | 293 LLVMValueRef scalar_pred = pred ? in emit_mask_scatter() local
300 if (scalar_pred) { in emit_mask_scatter()
303 real_val = lp_build_select(&bld->uint_elem_bld, scalar_pred, val, dst_val); in emit_mask_scatter()
|
| D | lp_bld_tgsi_soa.c | 645 LLVMValueRef scalar_pred = pred ? in emit_mask_scatter() local
652 if (scalar_pred) { in emit_mask_scatter()
655 real_val = lp_build_select(&bld->elem_bld, scalar_pred, val, dst_val); in emit_mask_scatter()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 4237 for (bool scalar_pred : {true, false}) { in TEST_F()
4241 if (scalar_pred) { in TEST_F()
4246 for (int i = 0; i < (scalar_pred ? 1 : 4); ++i) { in TEST_F()
|