Searched refs:relu_layer (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4034 nvinfer1::IActivationLayer* relu_layer = in ConvertRelu6() local
4037 TFTRT_RETURN_ERROR_IF_NULLPTR(relu_layer, node_def.name()); in ConvertRelu6()
4038 params->converter->SetLayerName(relu_layer, node_def, "activation"); in ConvertRelu6()
4044 params->converter->ProvideQuantizationRange(relu_layer->getOutput(0), 0.0f, in ConvertRelu6()
4050 params, 6.0f, relu_layer->getOutput(0)->getDimensions(), &const6_tensor)); in ConvertRelu6()
4057 *relu_layer->getOutput(0), *const6_tensor, in ConvertRelu6()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 324 name: "relu_layer"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_impl.py | 490 def relu_layer(x, weights, biases, name=None): function
|