Searched refs:second_weights (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | models.py | 289 second_weights = tf.compat.v1.get_variable(
301 filters=second_weights,
852 second_weights = tf.compat.v1.get_variable(
866 input=first_dropout, filters=second_weights,
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 383 const DeviceMemory<float>& second_weights, in DoSeparableConvolve() argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 407 const DeviceMemory<float>& second_weights, in DoSeparableConvolve() argument
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1383 const DeviceMemory<float>& second_weights,
|
| D | stream.cc | 680 const DeviceMemory<float> &second_weights, in ThenSeparableConvolve() argument
686 PARAM(depth_multiplier), PARAM(first_weights), PARAM(second_weights), in ThenSeparableConvolve()
692 first_weights, second_weights, convolution_descriptor, in ThenSeparableConvolve()
|
| D | stream.h | 430 const DeviceMemory<float> &second_weights,
|