| /external/tensorflow/tensorflow/python/ops/ |
| D | weights_broadcast_ops.py | 33 def _has_valid_dims(weights_shape, values_shape): argument
47 weights_rank, weights_shape, values_rank, values_shape): argument
|
| D | nn_batchnorm_test.py | 657 weights_shape, argument
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | create_im2col_arrays.cc | 40 const auto& weights_shape = weights_array.shape(); in ProcessConvOperator() local
|
| D | ensure_bias_vectors.cc | 31 const auto& weights_shape = model.GetArray(weights_name).shape(); in GetOutputDepthFromWeights() local
|
| D | convert_pure_conv_to_depthwise.cc | 101 const auto& weights_shape = weights_array.shape(); in Run() local
|
| D | shuffle_fc_weights.cc | 91 const Shape& weights_shape = weights_array.shape(); in Run() local
|
| D | fuse_binary_into_following_affine.cc | 70 const Shape& weights_shape = weights.shape(); in FuseAddOrSubParamsIntoFollowingAffine() local
|
| D | fuse_binary_into_preceding_affine.cc | 112 const Shape& weights_shape = weights.shape(); in FuseMulOrDivParamsIntoPrecedingAffine() local
|
| D | propagate_fixed_sizes.cc | 140 const auto& weights_shape = weights_array.shape(); in ProcessConvOperator() local
202 const auto& weights_shape = weights_array.shape(); in ProcessTransposeConvOperator() local
267 const auto& weights_shape = weights_array.shape(); in ProcessDepthwiseConvOperator() local
382 const auto& weights_shape = weights_array.shape(); in ProcessFullyConnectedOperator() local
1017 const auto& weights_shape = weights_array.shape(); in ProcessLstmCellOperator() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | fully_connected.h | 31 const float* input_data, const RuntimeShape& weights_shape, in FullyConnected()
170 const uint8* input_data, const RuntimeShape& weights_shape, in ShuffledFullyConnected()
|
| D | reference_ops.h | 1838 const float* prev_activ_data, const RuntimeShape& weights_shape, in LstmCell()
2041 const uint8* prev_activ_data_uint8, const RuntimeShape& weights_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 295 const RuntimeShape& weights_shape, in GEMVForLstmCell()
484 const RuntimeShape& weights_shape, const uint8* weights_data, in GEMVForLstmCellWithSymmetricRange()
775 const float* input_data, const RuntimeShape& weights_shape, in FullyConnected()
1633 const uint8* input_data, const RuntimeShape& weights_shape, in ShuffledFullyConnected()
3556 const float* prev_activ_data, const RuntimeShape& weights_shape, in LstmCell()
3678 const uint8* prev_activ_data_uint8, const RuntimeShape& weights_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 417 const RuntimeShape& weights_shape, in GEMVForLstmCell()
606 const RuntimeShape& weights_shape, const uint8* weights_data, in GEMVForLstmCellWithSymmetricRange()
897 const float* input_data, const RuntimeShape& weights_shape, in FullyConnected()
1689 const uint8* input_data, const RuntimeShape& weights_shape, in ShuffledFullyConnected()
3647 const float* prev_activ_data, const RuntimeShape& weights_shape, in LstmCell()
3794 const uint8* prev_activ_data_uint8, const RuntimeShape& weights_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2369 const auto& weights_shape = weights_array.shape(); in UndoWeightsShuffling() local
|
| D | export_tensorflow.cc | 1484 const auto& weights_shape = weights_array.shape(); in ConvertLstmCellOperator() local
|