Home
last modified time | relevance | path

Searched defs:values_vec (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/internal/
H A Dtfprof_tensor.cc45 std::vector<double> values_vec; in Build() local
57 std::vector<int64_t> values_vec; in Build() local
68 std::vector<tstring> values_vec; in Build() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsparse_matrix.h256 inline typename TTypes<T>::UnalignedVec values_vec(int batch) { in values_vec() function
266 inline typename TTypes<T>::UnalignedConstVec values_vec(int batch) const { in values_vec() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dreader_ops.cc123 std::vector<tstring> values_vec; in ComputeWithReader() local
H A Dsparse_utils_test.cc335 auto values_vec = output_values.vec<float>(); in GenerateRandomSparseTensor() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReduce.cuh537 const auto values_vec = memory::load_vector<input_vec_size>(data, idx); in input_vectorized_thread_reduce_impl() local