Searched refs:MakeReduceHlo (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.h | 151 StatusOr<HloInstruction*> MakeReduceHlo(HloInstruction* operand,
156 StatusOr<HloInstruction*> MakeReduceHlo(HloInstruction* operand,
|
| D | scatter_expander.cc | 211 MakeReduceHlo(valid_index, reduction_init, HloOpcode::kAnd, module)); in CheckIndexValidity()
|
| D | hlo_creation_utils.cc | 322 StatusOr<HloInstruction*> MakeReduceHlo(HloInstruction* operand, in MakeReduceHlo() function
347 StatusOr<HloInstruction*> MakeReduceHlo(HloInstruction* operand, in MakeReduceHlo() function
|
| D | algebraic_simplifier.cc | 4756 MakeReduceHlo(new_dot, init_value, reduce_dims, HloOpcode::kAdd)); in HandleReduce()
|