Searched defs:axis_id (Results 1 – 2 of 2) sorted by relevance
59 for(unsigned int axis_id = 1; axis_id < axis_dimension; ++axis_id) in softmax_layer_generic() local71 for(unsigned int axis_id = 0; axis_id < axis_dimension; ++axis_id) in softmax_layer_generic() local89 for(unsigned int axis_id = 0; axis_id < axis_dimension; ++axis_id) in softmax_layer_generic() local
5017 const int axis_id = node->inputs->data[1]; in AddOpsAndTensors() local