Home
last modified time | relevance | path

Searched defs:output_dim0 (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dconcat_split_util.h48 int64_t output_dim0 = 0; in Concat() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsegment_reduction_ops_test.cc199 Tensor output_dim0(DT_INT32, TensorShape({})); in SparseSegmentMeanGradHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
H A Dsegment_reduction_ops_test.py550 def _sparseSegmentReduceGrad(self, ygrad, indices, segment_ids, output_dim0, argument