Home
last modified time | relevance | path

Searched refs:second_weights (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/examples/speech_commands/
Dmodels.py279 second_weights = tf.get_variable(
290 second_conv = tf.nn.conv2d(max_pool, second_weights, [1, 1, 1, 1],
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h364 const DeviceMemory<float>& second_weights, in DoSeparableConvolve() argument
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h355 const DeviceMemory<float>& second_weights, in DoSeparableConvolve() argument
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1317 const DeviceMemory<float>& second_weights,
Dstream.cc799 const DeviceMemory<float> &second_weights, in ThenSeparableConvolve() argument
805 PARAM(depth_multiplier), PARAM(first_weights), PARAM(second_weights), in ThenSeparableConvolve()
812 depth_multiplier, first_weights, second_weights, in ThenSeparableConvolve()
Dstream.h399 const DeviceMemory<float> &second_weights,