Searched defs:weights_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | fully_connected_test.cc | 38 TfLiteIntArray* weights_dims = IntArrayFromInitializer(weights_dims_data); in TestFullyConnectedFloat() local
112 TfLiteIntArray* weights_dims = IntArrayFromInitializer(weights_dims_data); in TestFullyConnectedQuantized() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 328 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected()
346 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
428 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected()
823 const Dims<4>& weights_dims, const float* bias_data, in LstmCell()
847 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 394 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected()
412 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
474 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected()
522 const Dims<4>& weights_dims, const float* bias_data, in LstmCell()
546 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1830 for (auto weights_dims : std::vector<std::vector<int>>{{2, 1, 1}, {2}}) { in TestBinaryTensorOpWeightWithChannelWiseBroadcast() local
1891 const std::vector<int>& weights_dims, in TestBinaryTensorOpWeightFallback()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 932 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected()
950 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
1816 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected()
3766 const Dims<4>& weights_dims, const float* bias_data, in LstmCell()
4082 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell()
|