Searched refs:first_dim (Results 1 – 7 of 7) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorIO.h | 25 const Index first_dim = Eigen::internal::array_get<0>(tensor.dimensions()); in run() local
27 …ic, Dynamic, layout> > matrix(const_cast<Scalar*>(tensor.data()), first_dim, total_size/first_dim); in run()
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | debug.py | 101 first_dim = array_ops.shape(feature)[0]
103 batch_size = first_dim
105 size_checks.append(check_ops.assert_equal(batch_size, first_dim))
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 331 int64 first_dim = -1; in Compute() local
343 first_dim = element_shape.dim_size(0); in Compute()
365 bool check_dim = (first_dim == -1); in Compute()
366 int64 inferred_first_dim = first_dim; in Compute()
379 OP_REQUIRES(c, first_dim == -1 || first_dim == t.shape().dim_size(0), in Compute()
383 " Expected: ", first_dim, in Compute()
395 first_dim = inferred_first_dim; in Compute()
422 first_dim != -1) { in Compute()
423 dim = first_dim; in Compute()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | scatter_ops_test.py | 142 first_dim = 3 * size
143 indices = np.arange(first_dim)
172 old = _AsType(np.random.randn(*((first_dim,) + extra_shape)), vtype)
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 521 first_dim = column.num_buckets
523 first_dim = column.variable_shape.num_elements()
528 shape=(first_dim, self._units),
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 1345 first_dim = shape_tensor[axis:axis + ndims_mask].num_elements()
1347 tensor_shape.as_shape(shape_tensor[:axis]).concatenate([first_dim])
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1390 def _unflatten_first_dim(x, first_dim): argument
1393 new_shape = array_ops.concat([first_dim, [-1], old_shape[1:]], axis=0)
|