Home
last modified time | relevance | path

Searched refs:first_dim (Results 1 – 10 of 10) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorIO.h25 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/core/kernels/
Dlist_kernels.h307 int64 first_dim = -1; in Compute() local
319 first_dim = element_shape.dim_size(0); in Compute()
341 bool check_dim = (first_dim == -1); in Compute()
342 int64 inferred_first_dim = first_dim; in Compute()
355 OP_REQUIRES(c, first_dim == -1 || first_dim == t.shape().dim_size(0), in Compute()
359 " Expected: ", first_dim, in Compute()
371 first_dim = inferred_first_dim; in Compute()
397 first_dim != -1) { in Compute()
398 dim = first_dim; in Compute()
Dragged_tensor_to_tensor_op.cc140 Status CalculateOutputSize(INDEX_TYPE first_dim, OpKernelContext* c, in CalculateOutputSize() argument
172 (*result)[0] = first_dim; in CalculateOutputSize()
/external/tensorflow/tensorflow/python/kernel_tests/
Dscatter_ops_test.py142 first_dim = 3 * size
143 indices = np.arange(first_dim)
172 old = _AsType(np.random.randn(*((first_dim,) + extra_shape)), vtype)
Dtensor_array_ops_test.py1435 self.assertEqual(t.shape.as_list(), [first_dim, 3])
1440 first_dim = 1
1450 first_dim = None
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py302 first_dim = sum(int(v.shape[0]) for v in variables)
303 self._shape = tensor_shape.TensorShape([first_dim] + first_var.shape[1:])
/external/tensorflow/tensorflow/python/keras/layers/
Dtensorflow_op_layer_test.py585 first_dim = keras.Input(shape=(), dtype='int32')
590 out = inp[..., first_dim[0], slice_start[0]:slice_stop[0]:slice_stride[0]]
592 inputs=[inp, first_dim, slice_start, slice_stop, slice_stride],
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc275 const int64 first_dim = first_label - config.begin(); in EinsumDiagonalMask() local
276 mask = And(mask, Eq(Iota(builder, iota_shape, first_dim), in EinsumDiagonalMask()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py1734 def _unflatten_first_dim(x, first_dim): argument
1737 new_shape = array_ops.concat([first_dim, [-1], old_shape[1:]], axis=0)
3607 def _stack_tensor_list_shape(shape, first_dim): argument
3616 return first_dim
3622 lambda: array_ops.concat([first_dim, shape], axis=0))
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py1856 first_dim = shape_tensor[axis:axis + ndims_mask].num_elements()
1859 [first_dim]).concatenate(shape_tensor[axis + ndims_mask:]))