Searched refs:dynamic_dim_to_arg_num_map (Results 1 – 4 of 4) sorted by relevance
107 std::map<int32, int32> dynamic_dim_to_arg_num_map; member
1116 for (const auto& dim_and_arg_num : arg.dynamic_dim_to_arg_num_map) { in BuildArguments()
1825 args[0].dynamic_dim_to_arg_num_map.insert({1, 1}); in TEST_F()
356 .dynamic_dim_to_arg_num_map[padding_mapping.shape_index()] = in CompileTFFunctionToHlo()