Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClSoftmax.cpp76 TensorShape max_sum_shape = tmp_input_info.tensor_shape(); in configure() local
122 TensorShape max_sum_shape = src.tensor_shape(); in validate() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuSoftmax.cpp80 TensorShape max_sum_shape = tmp_input->tensor_shape(); in configure() local
132 TensorShape max_sum_shape = src->tensor_shape(); in validate() local