Searched defs:param_num (Results 1 – 5 of 5) sorted by relevance
136 for (int param_num = 0; param_num < inputs.size(); param_num++) { in ExecutionInputsFromScopedShapedBuffers() local
739 int64_t param_num; in ParseAliasing() local
222 int param_num = 0; in parsefmt() local
309 int param_num = use_tuple_arg ? 0 : update.input_index; in BuildComputation() local
2564 int param_num) { in XlaBroadcastKindSupported()2577 const DistributedTPURewritePass::ParameterInfo& params_info, int param_num, in EnableXlaParamBroadcast()