| /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));
159 Tensor weights_data(DT_FLOAT, TensorShape({2, 2}));
160 test::FillValues<float>(&weights_data, {1.0f, 2.0f, 0.3f, 0.4f});
162 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
|
| 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 | 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));
134 Tensor weights_data(DT_FLOAT, TensorShape({1, 2, 2, 2}));
135 test::FillValues<float>(&weights_data,
138 Const(root.WithOpName("weights_op"), Input::Initializer(weights_data));
|
| 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/contrib/lite/toco/graph_transformations/ |
| 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 | 104 float* const weights_data = weights_buffer.data.data(); local
152 weights_data[c * weights_per_depth + i] *=
155 weights_data[c * weights_per_depth + i] /=
164 weights_data[k * output_depth + c] *= operand_data[operand_channel];
166 weights_data[k * output_depth + c] /= operand_data[operand_channel];
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| embedding_lookup_sparse_test.cc | 55 std::initializer_list<float> weights_data) {
59 PopulateTensor(weights_, weights_data);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_fused_batch_norm_op.cc | 825 T* weights_data = reinterpret_cast<T*>(weights_m.get_data_handle()); variable
1173 T* weights_data = reinterpret_cast<T*>(weights_m.get_data_handle()); variable
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| reference_ops.h | 446 const float* weights_data,
468 weights_data[out_c * accum_depth + d];
483 const float* weights_data, const Dims<4>& weights_dims,
488 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data,
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| optimized_ops.h | 328 const float* weights_data,
347 MapAsMatrixWithFirstDimAsRows(weights_data, weights_dims);
360 const float* weights_data, const Dims<4>& weights_dims,
365 FullyConnected(input_data, input_dims, weights_data, weights_dims, bias_data,
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| export_tensorflow.cc | 1247 const float* weights_data = local
[all...] |
