Home
last modified time | relevance | path

Searched refs:weights_data (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h33 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight() argument
61 weights_data[pw1] * input_data[b * accum_depth + idx_1]; in FullyConnectedSparseWeight()
79 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4Impl() argument
100 weights_data, w1_segments, w1_indices, weights_shape.Dims(0), in FullyConnectedSparseWeight1x4Impl()
119 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4Task()
128 weights_data(weights_data), in FullyConnectedSparseWeight1x4Task()
139 sparsity, params, input_shape, input_data, weights_shape, weights_data, in Run()
150 const float* weights_data; member
167 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight1x4() argument
180 sparsity, params, input_shape, input_data, weights_shape, weights_data, in FullyConnectedSparseWeight1x4()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms_test.cc50 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldBatchNormsConv2D() local
51 test::FillValues<float>(&weights_data, in TestFoldBatchNormsConv2D()
54 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFoldBatchNormsConv2D()
101 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldBatchNormsDepthwiseConv2dNative() local
102 test::FillValues<float>(&weights_data, in TestFoldBatchNormsDepthwiseConv2dNative()
105 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFoldBatchNormsDepthwiseConv2dNative()
152 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldBatchNormsConv2DShared() local
153 test::FillValues<float>(&weights_data, in TestFoldBatchNormsConv2DShared()
156 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFoldBatchNormsConv2DShared()
210 Tensor weights_data(DT_FLOAT, TensorShape({2, 2})); in TestFoldBatchNormsMatMul() local
[all …]
Dfuse_convolutions_test.cc66 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFuseResizePadAndConv() local
67 test::FillValues<float>(&weights_data, in TestFuseResizePadAndConv()
70 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFuseResizePadAndConv()
116 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFuseResizeAndConv() local
117 test::FillValues<float>(&weights_data, in TestFuseResizeAndConv()
120 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFuseResizeAndConv()
168 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFusePadAndConv() local
169 test::FillValues<float>(&weights_data, in TestFusePadAndConv()
172 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFusePadAndConv()
Dfold_old_batch_norms_test.cc51 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldOldBatchNorms() local
52 test::FillValues<float>(&weights_data, in TestFoldOldBatchNorms()
55 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFoldOldBatchNorms()
135 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
136 test::FillValues<float>(&weights_data, in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
139 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
213 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldFusedBatchNorms() local
214 test::FillValues<float>(&weights_data, in TestFoldFusedBatchNorms()
217 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFoldFusedBatchNorms()
290 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
[all …]
Dround_weights_test.cc49 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 10})); in TestRoundWeights() local
51 &weights_data, in TestRoundWeights()
57 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestRoundWeights()
Dflatten_atrous_test.cc64 Tensor weights_data(DT_FLOAT, TensorShape({2, 2, 2, 1})); in TestFlattenAtrousConv() local
65 test::FillValues<float>(&weights_data, in TestFlattenAtrousConv()
68 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data)); in TestFlattenAtrousConv()
Dquantize_weights_test.cc50 Tensor weights_data(DT_FLOAT, weight_shape); in BuildGraphDef() local
51 test::FillValues<float>(&weights_data, weight_values); in BuildGraphDef()
53 Input::Initializer(weights_data)); in BuildGraphDef()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dshuffle_fc_weights.cc117 auto& weights_data = in Run() local
119 CHECK_EQ(rows * cols, weights_data.size()); in Run()
120 std::vector<uint8> shuffled_data(weights_data.size()); in Run()
125 const uint8* src_data_ptr = weights_data.data() + (r + i) * cols + c; in Run()
139 weights_data = std::move(shuffled_data); in Run()
Dfuse_binary_into_following_affine.cc73 const float* const weights_data = weights_buffer.data.data(); in FuseAddOrSubParamsIntoFollowingAffine() local
90 add_scalar_operand * weights_data[d * weights_per_depth + i]; in FuseAddOrSubParamsIntoFollowingAffine()
104 accumulation += add_scalar_operand * weights_data[k * output_depth + c]; in FuseAddOrSubParamsIntoFollowingAffine()
137 float* weights_data = in FuseMulOrDivParamsIntoFollowingAffine() local
142 weights_data[i] *= scalar_operand; in FuseMulOrDivParamsIntoFollowingAffine()
144 weights_data[i] /= scalar_operand; in FuseMulOrDivParamsIntoFollowingAffine()
Dfuse_binary_into_preceding_affine.cc130 float* const weights_data = weights_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
179 weights_data[c * weights_per_depth + i] *= in FuseMulOrDivParamsIntoPrecedingAffine()
182 weights_data[c * weights_per_depth + i] /= in FuseMulOrDivParamsIntoPrecedingAffine()
191 weights_data[k * output_depth + c] *= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
193 weights_data[k * output_depth + c] /= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed.h106 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in UploadWeights()
108 absl::MakeSpan(weights_data)); in UploadWeights()
114 desc.size = weights_data.size(); in UploadWeights()
115 desc.data = std::move(weights_data); in UploadWeights()
132 memcpy(desc.data.data(), weights_data.data() + sub_size * i, sub_size); in UploadWeights()
Dconv_powervr.h260 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in UploadWeights()
262 absl::MakeSpan(weights_data)); in UploadWeights()
272 desc.size = weights_data.size(); in UploadWeights()
273 desc.data = std::move(weights_data); in UploadWeights()
290 memcpy(desc.data.data(), weights_data.data() + sub_size * i, sub_size); in UploadWeights()
Dconv_weights_converter_test_util.cc55 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in ConvolutionWeightsConverterTest() local
57 absl::MakeSpan(weights_data)); in ConvolutionWeightsConverterTest()
74 memcpy(dst_tensors[i].data.data(), weights_data.data() + sub_size * i, in ConvolutionWeightsConverterTest()
81 memcpy(dst_tensors[0].data.data(), weights_data.data(), in ConvolutionWeightsConverterTest()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/
Dfully_connected.h28 const RuntimeShape& weights_shape, const float* weights_data, in FullyConnectedSparseWeight() argument
37 converter.SparseToDense(weights_data); in FullyConnectedSparseWeight()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dfully_connected_op_builder.cc54 const float* weights_data = GetTensorData<float>(weights_); in FillCoreMLWeights() local
55 std::copy(weights_data, weights_data + NumElements(weights_), in FillCoreMLWeights()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dfully_connected_slicing_test.cc40 const float input_max, const int* weights_dims_data, const T* weights_data, in TestFullyConnectedQuantized() argument
58 CreateQuantizedTensor(weights_data, weights_dims, "weights_tensor", 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
149 weights_data, weights_min, weights_max, bias_dims_data, bias_data, in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_sparse_test.cc59 std::initializer_list<float> weights_data) { in SetInput() argument
63 PopulateTensor(weights_, weights_data); in SetInput()
Dfully_connected_test.cc727 std::vector<float> weights_data(input_depth * output_depth); in SimpleTestQuantizedInt16OutputCase() local
728 for (auto& w : weights_data) { in SimpleTestQuantizedInt16OutputCase()
737 m.SetWeights<uint8_t>(weights_data); in SimpleTestQuantizedInt16OutputCase()
745 m.ShuffleAndSetWeights<uint8_t>(weights_data, input_depth, output_depth); in SimpleTestQuantizedInt16OutputCase()
779 input_data[b * input_depth + i] * weights_data[o * input_depth + i]; in SimpleTestQuantizedInt16OutputCase()
1188 const std::vector<T>& weights_data, in SparseFullyConnectedOpModel() argument
1200 AddConstSparseInput(weights, weights_data, symmetric_quantize_weights); in SparseFullyConnectedOpModel()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dfully_connected_test.cc266 const int* weights_dims_data, const float* weights_data, in TestFullyConnectedFloat() argument
281 CreateTensor(weights_data, weights_dims), in TestFullyConnectedFloat()
295 const int* weights_dims_data, const float* weights_data, in TestFullyConnectedQuantized() argument
314 CreateQuantizedTensor(weights_data, weights_quantized, weights_dims, in TestFullyConnectedQuantized()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc78 void Execute(const T* src_data, const U* weights_data, T* dst_data, in Execute() argument
90 static_cast<void*>(const_cast<U*>(weights_data)), *fwd_stream); in Execute()
109 static_cast<void*>(const_cast<U*>(weights_data))); in Execute()
422 const T* diff_dst_data, const U* weights_data, T* diff_src_data, in Execute() argument
438 static_cast<void*>(const_cast<U*>(weights_data)), *bwd_stream); in Execute()
457 static_cast<void*>(const_cast<U*>(weights_data))); in Execute()
817 U* weights_data = reinterpret_cast<U*>(weights.GetAllocatedBuffer()); in Compute() local
821 std::memcpy(weights_data, scale_tf, depth_ * sizeof(U)); in Compute()
822 std::memcpy(weights_data + depth_, shift_tf, depth_ * sizeof(U)); in Compute()
862 U* weights_op_data = weights_data; in Compute()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h645 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected() argument
665 MapAsMatrixWithLastDimAsRows(weights_data, weights_shape); in FullyConnected()
685 const float* weights_data, in FullyConnected() argument
696 DimsToShape(weights_dims), weights_data, in FullyConnected()
704 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected() argument
709 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
1157 const uint8* weights_data, uint8 weights_zero_point, in GEMVForLstmCell() argument
1192 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell()
1268 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell()
1345 const RuntimeShape& weights_shape, const uint8* weights_data, in GEMVForLstmCellWithSymmetricRange() argument
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h29 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected() argument
50 weights_data[out_c * accum_depth + d]; in FullyConnected()
Dlegacy_reference_ops.h408 const float* weights_data, in FullyConnected() argument
419 DimsToShape(weights_dims), weights_data, in FullyConnected()
427 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected() argument
432 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
571 const Dims<4>& prev_activ_dims, const float* weights_data, in LstmCell() argument
584 DimsToShape(weights_dims), weights_data, DimsToShape(bias_dims), in LstmCell()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc2429 auto& weights_data = in UndoWeightsShuffling() local
2437 CHECK_EQ(rows * cols, weights_data.size()); in UndoWeightsShuffling()
2439 std::vector<uint8> deshuffled_data(weights_data.size()); in UndoWeightsShuffling()
2440 uint8* shuffled_data_ptr = weights_data.data(); in UndoWeightsShuffling()
2457 CHECK_EQ(shuffled_data_ptr, weights_data.data() + rows * cols); in UndoWeightsShuffling()
2459 weights_data = std::move(deshuffled_data); in UndoWeightsShuffling()
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dceva_tflm_lib.h294 const float* weights_data, const int bias_shape, const float* bias_data,

12