Home
last modified time | relevance | path

Searched refs:create_auxiliary_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/
H A DGpuSoftmax.cpp139 …ITensorInfo *logits = sketch.implementation().create_auxiliary_tensor(src->clone()->set_… in create_op()
141 …ITensorInfo *sum = sketch.implementation().create_auxiliary_tensor(src->clone()->set_tensor_shape(… in create_op()
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/
H A DGpuWorkloadSketchImpl.h129 ITensorInfo *create_auxiliary_tensor(const ITensorInfo &tensor_info) in create_auxiliary_tensor() function