HomeSort by relevance Sort by last modified time
    Searched refs:num_dim (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/
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/
sparse_tensor.h 513 const int num_dim = input_tensor.shape().size(); local
520 CHECK(split_dim >= 0 && split_dim < num_dim) << "num_dim must be in "
522 << num_dim << ")";
534 TensorShape({num_values[i], num_dim}));
550 for (int j = 0; j < num_dim; ++j) {
  /external/tensorflow/tensorflow/core/ops/
sparse_ops.cc 236 DimensionHandle num_dim;
237 TF_RETURN_IF_ERROR(c->Merge(c->Dim(ind, 0), c->Dim(val, 0), &num_dim));
239 c->Add(output_row_count, num_dim, &output_row_count));

Completed in 234 milliseconds