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()
134 HloComputation::Builder max_computation("max"); in MaxComputationScalarF32() local136 auto param_lhs = max_computation.AddInstruction( in MaxComputationScalarF32()138 auto param_rhs = max_computation.AddInstruction( in MaxComputationScalarF32()140 max_computation.AddInstruction(HloInstruction::CreateBinary( in MaxComputationScalarF32()142 return max_computation.Build(); in MaxComputationScalarF32()
149 auto max_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F() local176 window, max_computation)); in TEST_F()