Searched refs:num_dim (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | data_format_ops.cc | 127 void ComputeDstIndex(int num_dim, Eigen::DSizes<Eigen::DenseIndex, 8>* dst) { in ComputeDstIndex() argument
132 for (int k = 0; k < num_dim; ++k) { in ComputeDstIndex()
133 (*dst)[i * num_dim + k] = j * num_dim + k; in ComputeDstIndex()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | clustering_ops_test.py | 157 num_dim = 100
163 [points_per_tile, num_dim]).astype(np.float32)
166 [num_centers, num_dim]).astype(np.float32)
|
| /external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/ |
| D | clustering_ops_test.py | 152 num_dim = 100
158 [points_per_tile, num_dim]).astype(np.float32)
161 [num_centers, num_dim]).astype(np.float32)
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 590 const int num_dim = input_tensor.shape().size(); in Split() local
599 if (!(split_dim >= 0 && split_dim < num_dim)) { in Split()
602 strings::StrCat("num_dim must be in the interval [0, ", num_dim, ")")); in Split()
615 TensorShape({num_values[i], num_dim})); in Split()
631 for (int j = 0; j < num_dim; ++j) { in Split()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 235 DimensionHandle num_dim; in __anona7ffb82a0a02() local
236 TF_RETURN_IF_ERROR(c->Merge(c->Dim(ind, 0), c->Dim(val, 0), &num_dim)); in __anona7ffb82a0a02()
238 c->Add(output_row_count, num_dim, &output_row_count)); in __anona7ffb82a0a02()
|