Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc215 const TfLiteFullyConnectedParams* matmul_params = in IsNodeSupportedByHexagon() local
219 IsActivationReluOrNone(matmul_params->activation) && in IsNodeSupportedByHexagon()
220 matmul_params->keep_num_dims == false && in IsNodeSupportedByHexagon()
221 matmul_params->weights_format == in IsNodeSupportedByHexagon()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_op_fused.cc119 MklDnnMatMulFwdParams matmul_params( in Compute() local
125 ExtendMklDnnMatMulFwdParams(ctx, matmul_params); in Compute()
127 MklDnnMatMulFwdPrimitiveFactory<T, T, T, T, T>::Get(matmul_params, 0); in Compute()