Searched refs:sparse_input_indices (Results 1 – 3 of 3) sorted by relevance
179 float FindSparseValue(const T1& sparse_input_indices, in FindSparseValue() argument185 int64 midi = internal::SubtleMustCopy(sparse_input_indices(mid, 0)); in FindSparseValue()186 int64 midj = internal::SubtleMustCopy(sparse_input_indices(mid, 1)); in FindSparseValue()
177 const Tensor& sparse_input_indices = context->input(2); in Compute() local182 data_set->set_input_tensors(input_data, sparse_input_indices, in Compute()278 const Tensor& sparse_input_indices = context->input(2); in Compute() local283 data_set->set_input_tensors(input_data, sparse_input_indices, in Compute()
242 const Tensor& sparse_input_indices = context->input(3); in Compute() local251 data_set->set_input_tensors(input_data, sparse_input_indices, in Compute()