Home
last modified time | relevance | path

Searched defs:output_memory_size (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DReduce.cuh1165 int64_t output_memory_size = iter.element_size(0); in gpu_reduce_kernel() local
1273 int64_t output_memory_size = iter.element_size(0); in jitted_gpu_reduce_kernel() local
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/grappler/costs/
Dop_performance_data.pb.h2255 inline int OpPerformance_OpMemory::output_memory_size() const { in output_memory_size() function