Searched refs:scalar_op (Results 1 – 3 of 3) sorted by relevance
279 auto scalar_op = b.AddInstruction( in HandleBatchGroupCount() local281 return computation_->parent()->AddEmbeddedComputation(b.Build(scalar_op)); in HandleBatchGroupCount()
89 auto scalar_op = b.AddInstruction(HloInstruction::CreateBinary( in GetOrCreateScalarAddComputation() local91 return computation_->parent()->AddEmbeddedComputation(b.Build(scalar_op)); in GetOrCreateScalarAddComputation()
410 auto scalar_op = b.AddInstruction(HloInstruction::CreateBinary( in GetOrCreateScalarAddComputation() local413 computation_->parent()->AddEmbeddedComputation(b.Build(scalar_op)); in GetOrCreateScalarAddComputation()