Searched defs:variable_index (Results 1 – 4 of 4) sorted by relevance
174 for (const auto variable_index : c10::irange(variable_count)) { in Reducer() local348 void Reducer::mark_variable_ready_dense(size_t variable_index) { in mark_variable_ready_dense()437 void Reducer::mark_variable_ready_sparse(size_t variable_index) { in mark_variable_ready_sparse()526 for (const auto variable_index : c10::irange(variable_count)) { in push_rebuilt_params_for_all_indices() local584 for (const auto variable_index : c10::irange(params_.size())) { in delay_all_reduce() local870 void Reducer::mark_variable_ready(size_t variable_index) { in mark_variable_ready()1001 for (const auto& variable_index : bucket.variable_indices) { in get_variables_for_bucket() local1100 const auto variable_index = bucket_indices[bucket_index].front(); in initialize_buckets() local1103 for (const auto variable_index : bucket_indices[bucket_index]) { in initialize_buckets() local1113 const auto variable_index = bucket_indices[bucket_index].front(); in initialize_buckets() local[all …]
394 unsigned int *variable_index, in fill_individual_variable()521 unsigned int *variable_index, in iterate_type_fill_variables()1026 unsigned *variable_index, in fill_block()1138 unsigned *variable_index, unsigned binding_offset, in fill_block_array()1196 unsigned variable_index = 0; in link_linked_shader_uniform_blocks() local
191 std::vector<int> variable_index; member246 std::vector<int> variable_index(arg_list.size(), -1); in BuildComputationInputs() local511 const int variable_index = input_buffers->variable_index.at(input_index); in AllocateOutputTensors() local548 const int variable_index = input_buffers->variable_index.at(input_index); in AllocateOutputTensors() local
2865 for (int variable_index = 0; variable_index < number_of_elements; variable_index++) in test_shader_compilation() local