Searched defs:output_dim0 (Results 1 – 2 of 2) sorted by relevance
129 Tensor output_dim0(DT_INT32, TensorShape({})); in SparseSegmentMeanGradHelper() local
48 int64 output_dim0 = 0; in Concat() local