Searched defs:output_dim0 (Results 1 – 3 of 3) sorted by relevance
129 Tensor output_dim0(DT_INT32, TensorShape({})); in SparseSegmentMeanGradHelper() local
1015 const Tensor& output_dim0 = context->input(3); in Compute() local
55 int64 output_dim0 = 0; in Concat() local