| /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
469 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/kernels/internal/optimized/sparse_ops/ |
| D | fully_connected.h | 33 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
79 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4Impl()
150 const float* weights_data; member
167 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4()
|
| /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 | 130 float* const weights_data = weights_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/ |
| D | fully_connected.h | 28 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | fully_connected_op_builder.cc | 54 const float* weights_data = GetTensorData<float>(weights_); in FillCoreMLWeights() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 59 std::initializer_list<float> weights_data) { in SetInput()
|
| D | fully_connected_test.cc | 727 std::vector<float> weights_data(input_depth * output_depth); in SimpleTestQuantizedInt16OutputCase() local
1188 const std::vector<T>& weights_data, in SparseFullyConnectedOpModel()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 78 void Execute(const T* src_data, const U* weights_data, T* dst_data, in Execute()
422 const T* diff_dst_data, const U* weights_data, T* diff_src_data, in Execute()
817 U* weights_data = reinterpret_cast<U*>(weights.GetAllocatedBuffer()); in Compute() local
1264 U* weights_data = weights_data_tf; in Compute() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/ |
| D | fully_connected_slicing_test.cc | 40 const float input_max, const int* weights_dims_data, const T* weights_data, in TestFullyConnectedQuantized()
138 const int8_t weights_data[] = {2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, in TF_LITE_MICRO_TEST() local
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | fully_connected_test.cc | 266 const int* weights_dims_data, const float* weights_data, in TestFullyConnectedFloat()
295 const int* weights_dims_data, const float* weights_data, in TestFullyConnectedQuantized()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_weights_converter_test_util.cc | 55 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in ConvolutionWeightsConverterTest() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | fully_connected.h | 29 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
|
| D | legacy_reference_ops.h | 408 const float* weights_data, in FullyConnected()
427 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
571 const Dims<4>& prev_activ_dims, const float* weights_data, in LstmCell()
|
| D | reference_ops.h | 443 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 645 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
685 const float* weights_data, in FullyConnected()
704 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected()
1157 const uint8* weights_data, uint8 weights_zero_point, in GEMVForLstmCell()
1345 const RuntimeShape& weights_shape, const uint8* weights_data, in GEMVForLstmCellWithSymmetricRange()
2983 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell()
3098 const Dims<4>& prev_activ_dims, const float* weights_data, in LstmCell()
|
| D | optimized_ops.h | 335 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
2778 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 2429 auto& weights_data = in UndoWeightsShuffling() local
|
| D | export_tensorflow.cc | 1518 const float* weights_data = in ConvertLstmCellOperator() local
|