Searched refs:add_binary (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 198 auto add_binary = [&](const Shape& shape, const HloOpcode opcode, in HandleBatchNormTraining() local
245 add_binary(operand_shape, HloOpcode::kMultiply, operand, operand); in HandleBatchNormTraining()
267 add_binary(feature_shape, HloOpcode::kMultiply, mean, mean); in HandleBatchNormTraining()
271 add_binary(feature_shape, HloOpcode::kSubtract, square_mean, mean_square); in HandleBatchNormTraining()
278 add_binary(operand_shape, HloOpcode::kAdd, var_broadcasted, epsilon); in HandleBatchNormTraining()
284 auto operand_minus_mean = add_binary(operand_shape, HloOpcode::kSubtract, in HandleBatchNormTraining()
288 auto normalized = add_binary(operand_shape, HloOpcode::kMultiply, in HandleBatchNormTraining()
292 auto scaled_normalized = add_binary(operand_shape, HloOpcode::kMultiply, in HandleBatchNormTraining()
296 auto shifted_normalized = add_binary(operand_shape, HloOpcode::kAdd, in HandleBatchNormTraining()
366 auto add_binary = [&](const Shape& shape, const HloOpcode opcode, in HandleBatchNormInference() local
[all …]
|