Searched defs:weights_dims (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | MKLDNNConversions.cpp | 284 const ideep::tensor::dims& weights_dims, in get_conv_transpose_expected_weights_desc()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 411 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected()
429 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
524 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected()
574 const Dims<4>& weights_dims, const float* bias_data, in LstmCell()
598 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 687 const Dims<4>& weights_dims, const float* bias_data, in FullyConnected()
705 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
2341 const uint8* shuffled_weights_data, const Dims<4>& weights_dims, in ShuffledFullyConnected()
3100 const Dims<4>& weights_dims, const float* bias_data, in LstmCell()
3411 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell()
|