| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| fold_batch_norms_test.cc | 50 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}));
51 test::FillValues<float>(&weights_data,
54 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
101 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}));
102 test::FillValues<float>(&weights_data,
105 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
152 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}));
153 test::FillValues<float>(&weights_data,
156 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
210 Tensor weights_data(DT_FLOAT, TensorShape({2, 2}))
[all...] |
| fuse_convolutions_test.cc | 66 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}));
67 test::FillValues<float>(&weights_data,
70 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
116 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}));
117 test::FillValues<float>(&weights_data,
120 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
168 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}));
169 test::FillValues<float>(&weights_data,
172 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
|
| round_weights_test.cc | 49 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 10}));
51 &weights_data,
57 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
|
| flatten_atrous_test.cc | 64 Tensor weights_data(DT_FLOAT, TensorShape({2, 2, 2, 1}));
65 test::FillValues<float>(&weights_data,
68 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
|
| fold_old_batch_norms_test.cc | 51 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}));
52 test::FillValues<float>(&weights_data,
55 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
135 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}));
136 test::FillValues<float>(&weights_data,
139 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
213 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}));
214 test::FillValues<float>(&weights_data,
217 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
290 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}))
[all...] |
| quantize_weights_test.cc | 50 Tensor weights_data(DT_FLOAT, weight_shape);
51 test::FillValues<float>(&weights_data, weight_values);
53 Input::Initializer(weights_data));
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| shuffle_fc_weights.cc | 117 auto& weights_data = local
119 CHECK_EQ(rows * cols, weights_data.size());
120 std::vector<uint8> shuffled_data(weights_data.size());
125 const uint8* src_data_ptr = weights_data.data() + (r + i) * cols + c;
139 weights_data = std::move(shuffled_data);
|
| fuse_binary_into_following_affine.cc | 73 const float* const weights_data = weights_buffer.data.data(); local
90 add_scalar_operand * weights_data[d * weights_per_depth + i];
104 accumulation += add_scalar_operand * weights_data[k * output_depth + c];
137 float* weights_data = local
142 weights_data[i] *= scalar_operand;
144 weights_data[i] /= scalar_operand;
|
| fuse_binary_into_preceding_affine.cc | 116 float* const weights_data = weights_buffer.data.data(); local
164 weights_data[c * weights_per_depth + i] *=
167 weights_data[c * weights_per_depth + i] /=
176 weights_data[k * output_depth + c] *= operand_data[operand_channel];
178 weights_data[k * output_depth + c] /= operand_data[operand_channel];
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| embedding_lookup_sparse_test.cc | 55 std::initializer_list<float> weights_data) {
59 PopulateTensor(weights_, weights_data);
|
| fully_connected_test.cc | 573 std::vector<float> weights_data(input_depth * output_depth);
574 for (auto& w : weights_data) {
583 m.SetWeights<uint8_t>(weights_data);
591 m.ShuffleAndSetWeights<uint8_t>(weights_data, input_depth, output_depth);
623 input_data[b * input_depth + i] * weights_data[o * input_depth + i];
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_fused_batch_norm_op.cc | 59 // weights_data: input data buffer of weights
63 void Execute(const T* src_data, const T* weights_data, T* dst_data,
71 static_cast<void*>(const_cast<T*>(weights_data)));
303 // weights_data: input data buffer of weights
307 const T* diff_dst_data, const T* weights_data, T* diff_src_data,
320 static_cast<void*>(const_cast<T*>(weights_data)));
622 T* weights_data = reinterpret_cast<T*>(weights.GetAllocatedBuffer()); variable
626 std::memcpy(weights_data, scale_tf, depth_ * sizeof(T));
627 std::memcpy(weights_data + depth_, shift_tf, depth_ * sizeof(T));
672 T* weights_op_data = weights_data;
1008 T* weights_data = weights_data_tf; variable
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| fully_connected.h | 32 const float* weights_data, const RuntimeShape& bias_shape,
53 weights_data[out_c * accum_depth + d];
|
| legacy_reference_ops.h | 393 const float* weights_data,
404 DimsToShape(weights_dims), weights_data,
412 const float* weights_data, const Dims<4>& weights_dims,
417 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data,
521 const Dims<4>& prev_activ_dims, const float* weights_data,
534 DimsToShape(weights_dims), weights_data, DimsToShape(bias_dims),
[all...] |
| reference_ops.h | [all...] |
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| fully_connected_test.cc | 30 std::initializer_list<float> weights_data,
48 CreateFloatTensor(weights_data, weights_dims, "weights_tensor"),
105 std::initializer_list<uint8_t> weights_data, float weights_min,
123 CreateQuantizedTensor(weights_data, weights_dims, "weights_tensor",
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| optimized_ops.h | 418 const uint8* weights_data, uint8 weights_zero_point,
453 const uint8* weights_ptr = weights_data + in + out * input_size;
529 const uint8* weights_ptr = weights_data + in + out * input_size;
606 const RuntimeShape& weights_shape, const uint8* weights_data,
655 optimized_ops_preload_l1_stream(weights_data + (out + 0) * input_size +
657 optimized_ops_preload_l1_stream(weights_data + (out + 1) * input_size +
659 optimized_ops_preload_l1_stream(weights_data + (out + 2) * input_size +
661 optimized_ops_preload_l1_stream(weights_data + (out + 3) * input_size +
668 optimized_ops_preload_l1_stream(weights_data + (out + 0) * input_size +
670 optimized_ops_preload_l1_stream(weights_data + (out + 1) * input_size
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| optimized_ops.h | 296 const uint8* weights_data, uint8 weights_zero_point,
331 const uint8* weights_ptr = weights_data + in + out * input_size;
407 const uint8* weights_ptr = weights_data + in + out * input_size;
484 const RuntimeShape& weights_shape, const uint8* weights_data,
533 optimized_ops_preload_l1_stream(weights_data + (out + 0) * input_size +
535 optimized_ops_preload_l1_stream(weights_data + (out + 1) * input_size +
537 optimized_ops_preload_l1_stream(weights_data + (out + 2) * input_size +
539 optimized_ops_preload_l1_stream(weights_data + (out + 3) * input_size +
546 optimized_ops_preload_l1_stream(weights_data + (out + 0) * input_size +
548 optimized_ops_preload_l1_stream(weights_data + (out + 1) * input_size
[all...] |
| legacy_optimized_ops.h | 327 const float* weights_data,
338 DimsToShape(weights_dims), weights_data,
346 const float* weights_data, const Dims<4>& weights_dims,
351 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data,
[all...] |
| /external/tensorflow/tensorflow/lite/toco/ |
| tooling_util.cc | 2367 auto& weights_data = local
[all...] |
| export_tensorflow.cc | 1488 const float* weights_data = local
[all...] |
