Home
last modified time | relevance | path

Searched defs:workspace_tensors (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEPooling3dLayer.cpp41 WorkspaceData<Tensor> workspace_tensors{}; member
H A DNEPoolingLayer.cpp42 WorkspaceData<Tensor> workspace_tensors{}; member
H A DNEAddMulAdd.cpp36 WorkspaceData<Tensor> workspace_tensors{}; member
H A DNESoftmaxLayer.cpp44 WorkspaceData<Tensor> workspace_tensors{}; member
H A DNEGEMMLowpMatrixMultiplyCore.cpp49 WorkspaceData<Tensor> workspace_tensors{}; member
H A DNEGEMMConvolutionLayer.cpp45 WorkspaceData<Tensor> workspace_tensors{}; member
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLSoftmaxLayer.cpp48 WorkspaceData<CLTensor> workspace_tensors{}; member
H A DCLGEMMLowpMatrixMultiplyCore.cpp54 WorkspaceData<CLTensor> workspace_tensors{}; member
H A DCLWinogradConvolutionLayer.cpp45 WorkspaceData<CLTensor> workspace_tensors{}; member
H A DCLGEMM.cpp50 WorkspaceData<CLTensor> workspace_tensors{}; member
H A DCLGEMMConvolutionLayer.cpp58 WorkspaceData<CLTensor> workspace_tensors{}; member
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc2177 std::vector<NodeBuilder::NodeOut>* workspace_tensors, in SetUpContiguousInputs()
2358 std::vector<NodeBuilder::NodeOut> workspace_tensors; in SetUpInputs() local
3555 std::vector<NodeBuilder::NodeOut> workspace_tensors; in RewriteNodeForJustOpNameChange() local