Searched refs:max_computation (Results 1 – 3 of 3) sorted by relevance
140 auto max_computation = CreateScalarMaxComputation(dtype, b); in MaxPool() local142 return ReduceWindow(operand, init_value, max_computation, kernel_size, in MaxPool()
1560 HloComputation::Builder max_computation("max"); in TEST_P() local1562 auto param_lhs = max_computation.AddInstruction( in TEST_P()1564 auto param_rhs = max_computation.AddInstruction( in TEST_P()1566 max_computation.AddInstruction(HloInstruction::CreateBinary( in TEST_P()1568 auto max_func = m_->AddEmbeddedComputation(max_computation.Build()); in TEST_P()1612 HloComputation::Builder max_computation("max"); in TEST_P() local1614 auto param_lhs = max_computation.AddInstruction( in TEST_P()1616 auto param_rhs = max_computation.AddInstruction( in TEST_P()1618 max_computation.AddInstruction(HloInstruction::CreateBinary( in TEST_P()1620 auto max_func = m_->AddEmbeddedComputation(max_computation.Build()); in TEST_P()
149 auto max_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local176 window, max_computation)); in TEST_F()