Home
last modified time | relevance | path

Searched refs:values_vec (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_tensor.cc45 std::vector<double> values_vec; in Build() local
47 GetValueVec<float, double>(&values_vec); in Build()
49 GetValueVec<double, double>(&values_vec); in Build()
51 BuildOutput<double>(0, 0, values_vec, &tfprof_tensor_pb_); in Build()
57 std::vector<int64> values_vec; in Build() local
59 GetValueVec<int32, int64>(&values_vec); in Build()
61 GetValueVec<int64, int64>(&values_vec); in Build()
63 BuildOutput<int64>(0, 0, values_vec, &tfprof_tensor_pb_); in Build()
68 std::vector<tstring> values_vec; in Build() local
69 GetValueVec<tstring, tstring>(&values_vec); in Build()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dreader_ops.cc123 std::vector<tstring> values_vec; in ComputeWithReader() local
124 values_vec.reserve(num_records); in ComputeWithReader()
127 reader->ReadUpTo(num_records, queue, &keys_vec, &values_vec, context); in ComputeWithReader()
132 OP_REQUIRES(context, num_actually_read == values_vec.size(), in ComputeWithReader()
149 values_t(i) = std::move(values_vec[i]); in ComputeWithReader()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dadd_op.cc113 a.values_vec<T>(i), a_dense_shape}; in operator ()()
115 b.values_vec<T>(i), b_dense_shape}; in operator ()()
138 a.values_vec<T>(i), a_dense_shape}; in operator ()()
140 b.values_vec<T>(i), b_dense_shape}; in operator ()()
168 a.values_vec<T>(i), a_dense_shape}; in operator ()()
170 b.values_vec<T>(i), b_dense_shape}; in operator ()()
172 c->values_vec<T>(i), c_dense_shape_t.vec<int64>()}; in operator ()()
Dsparse_mat_mul_op.cc276 csr_matrix.values_vec<T>(batch_index).data()); in GetSparseMatrixRef()
426 a_input_matrix->values_vec<T>(i), in Compute()
430 b_input_matrix->values_vec<T>(i), in Compute()
455 a_input_matrix->values_vec<T>(i), in Compute()
459 b_input_matrix->values_vec<T>(i), in Compute()
490 a_input_matrix->values_vec<T>(i), in Compute()
494 b_input_matrix->values_vec<T>(i), in Compute()
497 c.values_vec<T>(i), c_dense_shape}; in Compute()
Dtranspose_op.cc199 input_matrix.values_vec<T>(i), input_dense_shape}; in operator ()()
202 output_matrix->values_vec<T>(i), output_dense_shape}; in operator ()()
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
Dmat_mul_op.cc282 csr_matrix.values_vec<T>(batch_index).data() + row_offset); in GetSparseMatrixRef()
576 auto a_values = a_matrix->values_vec<T>(i); in Compute()
666 auto a_values = a_input_matrix->values_vec<T>(i); in Compute()
Dsparse_cholesky_op.cc121 input_matrix->values_vec<T>(batch_index).data()); in Compute()