Searched defs:first_dim (Results 1 – 7 of 7) sorted by relevance
25 const Index first_dim = Eigen::internal::array_get<0>(tensor.dimensions()); in run() local
212 def _equivalent_output_code_impl(self, size, first_dim=None, activation=True): argument
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
140 Status CalculateOutputSize(INDEX_TYPE first_dim, OpKernelContext* c, in CalculateOutputSize()
413 int64_t first_dim = -1; in Compute() local
1752 def _unflatten_first_dim(x, first_dim): argument3856 def _stack_tensor_list_shape(shape, first_dim): argument
306 const int64_t first_dim = first_label - config.begin(); in EinsumDiagonalMask() local