Searched defs:output_dim0 (Results 1 – 3 of 3) sorted by relevance
48 int64_t output_dim0 = 0; in Concat() local
131 Tensor output_dim0(DT_INT32, TensorShape({})); in SparseSegmentMeanGradHelper() local
522 def _sparseSegmentReduceGrad(self, ygrad, indices, segment_ids, output_dim0, argument