| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms_test.cc | 50 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldBatchNormsConv2D() local
101 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldBatchNormsDepthwiseConv2dNative() local
152 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldBatchNormsConv2DShared() local
210 Tensor weights_data(DT_FLOAT, TensorShape({2, 2})); in TestFoldBatchNormsMatMul() local
|
| D | fuse_convolutions_test.cc | 66 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFuseResizePadAndConv() local
116 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFuseResizeAndConv() local
168 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFusePadAndConv() local
|
| D | fold_old_batch_norms_test.cc | 51 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldOldBatchNorms() local
135 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
213 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldFusedBatchNorms() local
290 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
468 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
|
| D | round_weights_test.cc | 49 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 10})); in TestRoundWeights() local
|
| D | flatten_atrous_test.cc | 64 Tensor weights_data(DT_FLOAT, TensorShape({2, 2, 2, 1})); in TestFlattenAtrousConv() local
|
| D | quantize_weights_test.cc | 50 Tensor weights_data(DT_FLOAT, weight_shape); in BuildGraphDef() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | fuse_binary_into_following_affine.cc | 73 const float* const weights_data = weights_buffer.data.data(); in FuseAddOrSubParamsIntoFollowingAffine() local
137 float* weights_data = in FuseMulOrDivParamsIntoFollowingAffine() local
|
| D | shuffle_fc_weights.cc | 117 auto& weights_data = in Run() local
|
| D | fuse_binary_into_preceding_affine.cc | 116 float* const weights_data = weights_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | fully_connected_test.cc | 30 std::initializer_list<float> weights_data, in TestFullyConnectedFloat()
105 std::initializer_list<uint8_t> weights_data, float weights_min, in TestFullyConnectedQuantized()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 55 std::initializer_list<float> weights_data) { in SetInput()
|
| D | fully_connected_test.cc | 573 std::vector<float> weights_data(input_depth * output_depth); in SimpleTestQuantizedInt16OutputCase() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_fused_batch_norm_op.cc | 63 void Execute(const T* src_data, const T* weights_data, T* dst_data, in Execute()
307 const T* diff_dst_data, const T* weights_data, T* diff_src_data, in Execute()
622 T* weights_data = reinterpret_cast<T*>(weights.GetAllocatedBuffer()); in Compute() local
1008 T* weights_data = weights_data_tf; in Compute() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | fully_connected.h | 32 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
|
| D | legacy_reference_ops.h | 393 const float* weights_data, in FullyConnected()
412 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
521 const Dims<4>& prev_activ_dims, const float* weights_data, in LstmCell()
|
| D | reference_ops.h | 1839 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 418 const uint8* weights_data, uint8 weights_zero_point, in GEMVForLstmCell()
606 const RuntimeShape& weights_shape, const uint8* weights_data, in GEMVForLstmCellWithSymmetricRange()
898 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
931 const float* weights_data, in FullyConnected()
950 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
3648 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell()
3765 const Dims<4>& prev_activ_dims, const float* weights_data, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 327 const float* weights_data, in FullyConnected()
346 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
822 const Dims<4>& prev_activ_dims, const float* weights_data, in LstmCell()
|
| D | optimized_ops.h | 296 const uint8* weights_data, uint8 weights_zero_point, in GEMVForLstmCell()
484 const RuntimeShape& weights_shape, const uint8* weights_data, in GEMVForLstmCellWithSymmetricRange()
776 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
3557 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2367 auto& weights_data = in UndoWeightsShuffling() local
|
| D | export_tensorflow.cc | 1488 const float* weights_data = in ConvertLstmCellOperator() local
|