Searched refs:dynamic_size_param_index (Results 1 – 3 of 3) sorted by relevance
1048 int dynamic_size_param_index = arg_to_inputs.at(dim_and_arg_num.second); in BuildArguments() local1050 << dynamic_size_param_index; in BuildArguments()1053 /*dynamic_size_param_num=*/0, {dynamic_size_param_index}, in BuildArguments()1089 int dynamic_size_param_index = arg_to_inputs.at(dim_and_arg_num.second); in BuildArguments() local1091 /*dynamic_size_param_num=*/dynamic_size_param_index, {}, in BuildArguments()
298 ShapeIndex dynamic_size_param_index, in SetDynamicBinding() argument329 dynamic_size_param_index}, in SetDynamicBinding()
328 ShapeIndex dynamic_size_param_index,