| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | fully_connected_test.cc | 38 TfLiteIntArray* weights_dims = IntArrayFromInitializer(weights_dims_data); in TestFullyConnectedFloat() local
48 CreateFloatTensor(weights_data, weights_dims, "weights_tensor"), in TestFullyConnectedFloat()
112 TfLiteIntArray* weights_dims = IntArrayFromInitializer(weights_dims_data); in TestFullyConnectedQuantized() local
123 CreateQuantizedTensor(weights_data, weights_dims, "weights_tensor", in TestFullyConnectedQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 328 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected() argument
338 DimsToShape(weights_dims), weights_data, in FullyConnected()
346 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected() argument
351 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
428 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected() argument
441 DimsToShape(weights_dims), shuffled_weights_data, in ShuffledFullyConnected()
823 const Dims<4>& weights_dims, const float* bias_data, in LstmCell() argument
835 DimsToShape(weights_dims), weights_data, DimsToShape(bias_dims), in LstmCell()
847 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell() argument
864 DimsToShape(weights_dims), weights_data_uint8, DimsToShape(bias_dims), 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() argument
404 DimsToShape(weights_dims), weights_data, in FullyConnected()
412 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected() argument
417 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
474 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected() argument
487 DimsToShape(weights_dims), shuffled_weights_data, in ShuffledFullyConnected()
522 const Dims<4>& weights_dims, const float* bias_data, in LstmCell() argument
534 DimsToShape(weights_dims), weights_data, DimsToShape(bias_dims), in LstmCell()
546 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell() argument
563 DimsToShape(weights_dims), weights_data_uint8, DimsToShape(bias_dims), 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
1834 test->AddTestWeights<CType>("weights", weights_dims, weights); in TestBinaryTensorOpWeightWithChannelWiseBroadcast()
1849 if (weights_dims.size() == 1) { in TestBinaryTensorOpWeightWithChannelWiseBroadcast()
1891 const std::vector<int>& weights_dims, in TestBinaryTensorOpWeightFallback() argument
1898 const size_t num_weights = TrtDimsNumElements(GetTestDims(weights_dims)); in TestBinaryTensorOpWeightFallback()
1906 "weights", /*dims=*/weights_dims, in TestBinaryTensorOpWeightFallback()
1920 for (int i = expected_output_dims.size() - 1, j = weights_dims.size() - 1; in TestBinaryTensorOpWeightFallback()
1923 expected_output_dims[i] = weights_dims[j]; 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() argument
942 DimsToShape(weights_dims), weights_data, in FullyConnected()
950 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected() argument
955 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
1816 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected() argument
1829 DimsToShape(weights_dims), shuffled_weights_data, in ShuffledFullyConnected()
3766 const Dims<4>& weights_dims, const float* bias_data, in LstmCell() argument
3778 DimsToShape(weights_dims), weights_data, DimsToShape(bias_dims), in LstmCell()
4082 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell() argument
4099 DimsToShape(weights_dims), weights_data_uint8, DimsToShape(bias_dims), in LstmCell()
|