/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/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/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/toco/ |
export_tensorflow.cc | 1247 const float* weights_data = local [all...] |