Home
last modified time | relevance | path

Searched refs:variable_index (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.cpp174 for (const auto variable_index : c10::irange(variable_count)) { in Reducer() local
175 auto& variable = params_[variable_index]; in Reducer()
189 [this, variable_index]( in Reducer()
196 this->autograd_hook(variable_index); in Reducer()
212 gradAccToVariableMap_[grad_accumulator.get()] = variable_index; in Reducer()
215 numGradHooksTriggeredMap_[variable_index] = 0; in Reducer()
221 grad_accumulators_[variable_index] == nullptr, in Reducer()
225 variable_index)); in Reducer()
227 grad_accumulators_[variable_index] = std::move(grad_accumulator); in Reducer()
348 void Reducer::mark_variable_ready_dense(size_t variable_index) { in mark_variable_ready_dense() argument
[all …]
H A Dreducer.hpp272 void mark_variable_ready_dense(size_t variable_index);
274 void mark_variable_ready_sparse(size_t variable_index);
276 void mark_variable_ready(size_t variable_index);
/aosp_15_r20/external/mesa3d/src/compiler/glsl/
H A Dgl_nir_link_uniform_blocks.c394 unsigned int *variable_index, in fill_individual_variable() argument
403 struct gl_uniform_buffer_variable *v = &variables[*variable_index]; in fill_individual_variable()
495 (*variable_index)++; in fill_individual_variable()
521 unsigned int *variable_index, in iterate_type_fill_variables() argument
592 variable_index, offset, buffer_size, prog, in iterate_type_fill_variables()
597 variable_index, offset, buffer_size, in iterate_type_fill_variables()
1026 unsigned *variable_index, in fill_block() argument
1055 block->Uniforms = &variables[*variable_index]; in fill_block()
1076 unsigned old_variable_index = *variable_index; in fill_block()
1084 …pe_fill_variables(mem_ctx, &ifc_name_dup, ifc_name_length, consts, type, variables, variable_index, in fill_block()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_execute_op.cc191 std::vector<int> variable_index; member
246 std::vector<int> variable_index(arg_list.size(), -1); in BuildComputationInputs() local
252 variable_index[i] = variables.size(); in BuildComputationInputs()
348 int vi = variable_index[i]; in BuildComputationInputs()
355 input_buffers->variable_index = std::move(variable_index); in BuildComputationInputs()
511 const int variable_index = input_buffers->variable_index.at(input_index); in AllocateOutputTensors() local
515 if (variable_index < 0) { in AllocateOutputTensors()
548 const int variable_index = input_buffers->variable_index.at(input_index); in AllocateOutputTensors() local
549 if (variable_index >= 0) { in AllocateOutputTensors()
552 VariableInfo& var = input_buffers->variables[variable_index]; in AllocateOutputTensors()
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dshared_variable_creator.py79 variable_index = variable_scope_access_index.get(canonical_name, 0)
80 v = shared_variable_store[canonical_name][variable_index]
83 variable_scope_access_index[canonical_name] = variable_index + 1
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/
H A Dxla_ops.cc335 for (const auto& [variable_index, variable_tensor] : variables_snapshot) { in ComputeAsync()
336 snapshot_ptrs.emplace(variable_index, variable_tensor.has_value() in ComputeAsync()
602 for (const auto& [variable_index, variable_tensor] : in Compute()
604 snapshot_ptrs.emplace(variable_index, variable_tensor.has_value() in Compute()
/aosp_15_r20/external/deqp/external/openglcts/modules/gles31/
H A Des31cArrayOfArraysTests.cpp2865 for (int variable_index = 0; variable_index < number_of_elements; variable_index++) in test_shader_compilation() local
2867 for (int value_index = variable_index; value_index < number_of_elements; value_index++) in test_shader_compilation()
2872 if (variable_index != value_index) in test_shader_compilation()
2874 …shader_source += " " + variable_basenames[variable_index] + " = " + variable_basenames[value_in… in test_shader_compilation()