Home
last modified time | relevance | path

Searched refs:shuffled_weights_data (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h171 const uint8* shuffled_weights_data, const RuntimeShape& bias_shape, in ShuffledFullyConnected() argument
232 reinterpret_cast<const int8*>(shuffled_weights_data); in ShuffledFullyConnected()
272 reinterpret_cast<const int8*>(shuffled_weights_data); in ShuffledFullyConnected()
Dlegacy_reference_ops.h474 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected() argument
487 DimsToShape(weights_dims), shuffled_weights_data, in ShuffledFullyConnected()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h1375 const int8* shuffled_weights_data, int batches, int output_depth, in ShuffledFullyConnectedWorkerImpl() argument
1379 const int8* shuffled_weights_ptr = shuffled_weights_data; in ShuffledFullyConnectedWorkerImpl()
1555 reinterpret_cast<const int8*>(shuffled_weights_data); in ShuffledFullyConnectedWorkerImpl()
1595 reinterpret_cast<const int8*>(shuffled_weights_data); in ShuffledFullyConnectedWorkerImpl()
1651 const int8* shuffled_weights_data, in ShuffledFullyConnectedWorkerTask()
1658 shuffled_weights_data_(shuffled_weights_data), in ShuffledFullyConnectedWorkerTask()
1690 const uint8* shuffled_weights_data, const RuntimeShape& bias_shape, in ShuffledFullyConnected() argument
1723 reinterpret_cast<const int8*>(shuffled_weights_data); in ShuffledFullyConnected()
1816 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected() argument
1829 DimsToShape(weights_dims), shuffled_weights_data, in ShuffledFullyConnected()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1319 const int8* shuffled_weights_data, int batches, int output_depth, in ShuffledFullyConnectedWorkerImpl() argument
1323 const int8* shuffled_weights_ptr = shuffled_weights_data; in ShuffledFullyConnectedWorkerImpl()
1499 reinterpret_cast<const int8*>(shuffled_weights_data); in ShuffledFullyConnectedWorkerImpl()
1539 reinterpret_cast<const int8*>(shuffled_weights_data); in ShuffledFullyConnectedWorkerImpl()
1595 const int8* shuffled_weights_data, in ShuffledFullyConnectedWorkerTask()
1602 shuffled_weights_data_(shuffled_weights_data), in ShuffledFullyConnectedWorkerTask()
1634 const uint8* shuffled_weights_data, const RuntimeShape& bias_shape, in ShuffledFullyConnected() argument
1667 reinterpret_cast<const int8*>(shuffled_weights_data); in ShuffledFullyConnected()
Dlegacy_optimized_ops.h428 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected() argument
441 DimsToShape(weights_dims), shuffled_weights_data, in ShuffledFullyConnected()