Home
last modified time | relevance | path

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

/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.cc116 float* const weights_data = weights_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine() local
164 weights_data[c * weights_per_depth + i] *= in FuseMulOrDivParamsIntoPrecedingAffine()
167 weights_data[c * weights_per_depth + i] /= in FuseMulOrDivParamsIntoPrecedingAffine()
176 weights_data[k * output_depth + c] *= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
178 weights_data[k * output_depth + c] /= operand_data[operand_channel]; in FuseMulOrDivParamsIntoPrecedingAffine()
/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_sparse_test.cc55 std::initializer_list<float> weights_data) { in SetInput() argument
59 PopulateTensor(weights_, weights_data); in SetInput()
Dfully_connected_test.cc573 std::vector<float> weights_data(input_depth * output_depth); in SimpleTestQuantizedInt16OutputCase() local
574 for (auto& w : weights_data) { in SimpleTestQuantizedInt16OutputCase()
583 m.SetWeights<uint8_t>(weights_data); in SimpleTestQuantizedInt16OutputCase()
591 m.ShuffleAndSetWeights<uint8_t>(weights_data, input_depth, output_depth); in SimpleTestQuantizedInt16OutputCase()
623 input_data[b * input_depth + i] * weights_data[o * input_depth + i]; in SimpleTestQuantizedInt16OutputCase()
/external/tensorflow/tensorflow/core/kernels/
Dmkl_fused_batch_norm_op.cc63 void Execute(const T* src_data, const T* weights_data, T* dst_data, in Execute() argument
71 static_cast<void*>(const_cast<T*>(weights_data))); in Execute()
307 const T* diff_dst_data, const T* weights_data, T* diff_src_data, in Execute() argument
320 static_cast<void*>(const_cast<T*>(weights_data))); in Execute()
622 T* weights_data = reinterpret_cast<T*>(weights.GetAllocatedBuffer()); in Compute() local
626 std::memcpy(weights_data, scale_tf, depth_ * sizeof(T)); in Compute()
627 std::memcpy(weights_data + depth_, shift_tf, depth_ * sizeof(T)); in Compute()
672 T* weights_op_data = weights_data; in Compute()
1008 T* weights_data = weights_data_tf; in Compute() local
1013 weights_data, diff_src_data, diff_weights_data); in Compute()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h418 const uint8* weights_data, uint8 weights_zero_point, in GEMVForLstmCell() argument
453 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell()
529 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell()
606 const RuntimeShape& weights_shape, const uint8* weights_data, in GEMVForLstmCellWithSymmetricRange() argument
655 optimized_ops_preload_l1_stream(weights_data + (out + 0) * input_size + in GEMVForLstmCellWithSymmetricRange()
657 optimized_ops_preload_l1_stream(weights_data + (out + 1) * input_size + in GEMVForLstmCellWithSymmetricRange()
659 optimized_ops_preload_l1_stream(weights_data + (out + 2) * input_size + in GEMVForLstmCellWithSymmetricRange()
661 optimized_ops_preload_l1_stream(weights_data + (out + 3) * input_size + in GEMVForLstmCellWithSymmetricRange()
668 optimized_ops_preload_l1_stream(weights_data + (out + 0) * input_size + in GEMVForLstmCellWithSymmetricRange()
670 optimized_ops_preload_l1_stream(weights_data + (out + 1) * input_size + in GEMVForLstmCellWithSymmetricRange()
[all …]
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dfully_connected_test.cc30 std::initializer_list<float> weights_data, in TestFullyConnectedFloat() argument
48 CreateFloatTensor(weights_data, weights_dims, "weights_tensor"), in TestFullyConnectedFloat()
105 std::initializer_list<uint8_t> weights_data, float weights_min, in TestFullyConnectedQuantized() argument
123 CreateQuantizedTensor(weights_data, weights_dims, "weights_tensor", in TestFullyConnectedQuantized()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h32 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected() argument
53 weights_data[out_c * accum_depth + d]; in FullyConnected()
Dlegacy_reference_ops.h393 const float* weights_data, in FullyConnected() argument
404 DimsToShape(weights_dims), weights_data, in FullyConnected()
412 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected() argument
417 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
521 const Dims<4>& prev_activ_dims, const float* weights_data, in LstmCell() argument
534 DimsToShape(weights_dims), weights_data, DimsToShape(bias_dims), in LstmCell()
Dreference_ops.h1839 const float* weights_data, const RuntimeShape& unextended_bias_shape, in LstmCell() argument
1917 weights_data, bias_shape, bias_data, activ_temp_shape, in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h296 const uint8* weights_data, uint8 weights_zero_point, in GEMVForLstmCell() argument
331 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell()
407 const uint8* weights_ptr = weights_data + in + out * input_size; in GEMVForLstmCell()
484 const RuntimeShape& weights_shape, const uint8* weights_data, in GEMVForLstmCellWithSymmetricRange() argument
533 optimized_ops_preload_l1_stream(weights_data + (out + 0) * input_size + in GEMVForLstmCellWithSymmetricRange()
535 optimized_ops_preload_l1_stream(weights_data + (out + 1) * input_size + in GEMVForLstmCellWithSymmetricRange()
537 optimized_ops_preload_l1_stream(weights_data + (out + 2) * input_size + in GEMVForLstmCellWithSymmetricRange()
539 optimized_ops_preload_l1_stream(weights_data + (out + 3) * input_size + in GEMVForLstmCellWithSymmetricRange()
546 optimized_ops_preload_l1_stream(weights_data + (out + 0) * input_size + in GEMVForLstmCellWithSymmetricRange()
548 optimized_ops_preload_l1_stream(weights_data + (out + 1) * input_size + in GEMVForLstmCellWithSymmetricRange()
[all …]
Dlegacy_optimized_ops.h327 const float* weights_data, in FullyConnected() argument
338 DimsToShape(weights_dims), weights_data, in FullyConnected()
346 const float* weights_data, const Dims<4>& weights_dims, in FullyConnected() argument
351 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data, in FullyConnected()
822 const Dims<4>& prev_activ_dims, const float* weights_data, in LstmCell() argument
835 DimsToShape(weights_dims), weights_data, DimsToShape(bias_dims), in LstmCell()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc2367 auto& weights_data = in UndoWeightsShuffling() local
2375 CHECK_EQ(rows * cols, weights_data.size()); in UndoWeightsShuffling()
2377 std::vector<uint8> deshuffled_data(weights_data.size()); in UndoWeightsShuffling()
2378 uint8* shuffled_data_ptr = weights_data.data(); in UndoWeightsShuffling()
2395 CHECK_EQ(shuffled_data_ptr, weights_data.data() + rows * cols); in UndoWeightsShuffling()
2397 weights_data = std::move(deshuffled_data); in UndoWeightsShuffling()
Dexport_tensorflow.cc1488 const float* weights_data = in ConvertLstmCellOperator() local
1490 ConvertFloatTensorConst(weights_output, weights_shape, weights_data, in ConvertLstmCellOperator()