Searched refs:shuffled_weights_data (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | fully_connected.h | 171 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()
|
D | legacy_reference_ops.h | 474 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/ |
D | optimized_ops.h | 1375 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/ |
D | optimized_ops.h | 1319 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()
|
D | legacy_optimized_ops.h | 428 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected() argument 441 DimsToShape(weights_dims), shuffled_weights_data, in ShuffledFullyConnected()
|