Searched refs:max_computation (Results 1 – 3 of 3) sorted by relevance
141 auto max_computation = CreateScalarMaxComputation(dtype, b); in MaxPool() local143 return ReduceWindow(operand, init_value, max_computation, kernel_size, in MaxPool()
150 auto max_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local177 window, max_computation)); in TEST_F()
159 HloComputation::Builder max_computation("max"); in MaxComputationScalarF32() local161 auto param_lhs = max_computation.AddInstruction( in MaxComputationScalarF32()163 auto param_rhs = max_computation.AddInstruction( in MaxComputationScalarF32()165 max_computation.AddInstruction(HloInstruction::CreateBinary( in MaxComputationScalarF32()167 return max_computation.Build(); in MaxComputationScalarF32()