Searched defs:first_dim (Results 1 – 6 of 6) sorted by relevance
25 const Index first_dim = Eigen::internal::array_get<0>(tensor.dimensions()); in run() local
140 Status CalculateOutputSize(INDEX_TYPE first_dim, OpKernelContext* c, in CalculateOutputSize()
413 int64_t first_dim = -1; in Compute() local
253 const int64_t first_dim = GetFirstChosenSpatialDim(convolution); in GetChosenSpatialDims() local2706 const int64_t first_dim = GetFirstChosenSpatialDim(convolution); in PropagateOnConv() local3778 const int64_t first_dim = GetFirstChosenSpatialDim(convolution); in PerformSpaceToBatchOnConvolution() local
306 const int64_t first_dim = first_label - config.begin(); in EinsumDiagonalMask() local
1752 def _unflatten_first_dim(x, first_dim): argument3856 def _stack_tensor_list_shape(shape, first_dim): argument