Searched refs:call_param (Results 1 – 3 of 3) sorted by relevance
635 fake_gen_each_reply_with_call_param = proc do |msgs, call_param|636 call_param.merge_metadata_to_send(@server_to_client_metadata)637 call_param.send_initial_metadata
1594 const HloInstruction* call_param = callee->parameter_instruction(i); in BuildColocatedBufferSets() local1600 AddBufferToColocatedSet(call_param, index, points_to_analysis, in BuildColocatedBufferSets()
1212 auto call_param = call_builder.AddInstruction( in TEST_F() local1215 HloInstruction::CreateUnary(vec_shape, HloOpcode::kExp, call_param)); in TEST_F()1244 auto& call_param_alloc = GetTopLevelAllocation(*assignment, call_param); in TEST_F()