Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSegmentReduce.cu283 auto zeros_shape = offsets.sizes().vec(); in _segment_reduce_lengths_offsets_backward_cuda_kernel() local
411 auto zeros_shape = offsets.sizes().vec(); in _segment_reduce_lengths_offsets_cuda_kernel() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_tensor_to_variant_op.cc272 TensorShape zeros_shape = dense_values_shape; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsparsify_gather_test.cc84 NodeDef* zeros_shape; in TestSinglePartition() local