Home
last modified time | relevance | path

Searched refs:SelectSoftmax (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.h78 void SelectSoftmax(const BHWC& shape, const OperationDef& op_def,
Dsimple_selectors.cc170 void SelectSoftmax(const BHWC& shape, const OperationDef& op_def, in SelectSoftmax() function
Doperation_selector.cc485 SelectSoftmax(inputs[0]->tensor.shape, op_def, gpu_op); in GPUOperationFromNode()