Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc296 auto shifted_normalized = add_binary(operand_shape, HloOpcode::kAdd, in HandleBatchNormTraining() local
299 auto tuple = HloInstruction::CreateTuple({shifted_normalized, mean, var}); in HandleBatchNormTraining()
404 auto shifted_normalized = HloInstruction::CreateBinary( in HandleBatchNormInference() local
424 shifted_normalized->set_sharding(sharding); in HandleBatchNormInference()
427 ReplaceWithNewInstruction(batch_norm, std::move(shifted_normalized))); in HandleBatchNormInference()