Home
last modified time | relevance | path

Searched refs:num_state_variables (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_embedding_load_retrieve_ops.cc268 std::vector<int> & num_state_variables) { in GetStateVariables() argument
291 num_state_variables.resize(num_tables); in GetStateVariables()
305 num_state_variables[table_id] = state_variable_specs.size(); in GetStateVariables()
336 std::vector<int> num_state_variables; in Compute() local
338 GetStateVariables(ctx, state_variable_vector, num_state_variables); in Compute()
364 for (int i = 0; i < num_state_variables[table_id]; ++i) { in Compute()
H A Dtpu_embedding_load_retrieve_ops.h85 std::vector<int>& num_state_variables);