Searched refs:first_dim (Results 1 – 10 of 10) 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/core/kernels/ |
D | list_kernels.h | 307 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()
|
D | ragged_tensor_to_tensor_op.cc | 140 Status CalculateOutputSize(INDEX_TYPE first_dim, OpKernelContext* c, in CalculateOutputSize() argument 172 (*result)[0] = first_dim; in CalculateOutputSize()
|
/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)
|
D | tensor_array_ops_test.py | 1435 self.assertEqual(t.shape.as_list(), [first_dim, 3]) 1440 first_dim = 1 1450 first_dim = None
|
/external/tensorflow/tensorflow/python/distribute/ |
D | sharded_variable.py | 302 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/ |
D | tensorflow_op_layer_test.py | 585 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/ |
D | matrix.cc | 275 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/ |
D | pfor.py | 1734 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/ |
D | array_ops.py | 1856 first_dim = shape_tensor[axis:axis + ndims_mask].num_elements() 1859 [first_dim]).concatenate(shape_tensor[axis + ndims_mask:]))
|