| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| ensure_bias_vectors.cc | 30 const string& weights_name = op.inputs[1]; local
31 const auto& weights_shape = model.GetArray(weights_name).shape();
48 const string& weights_name = op->inputs[1]; local
49 if (!model->GetArray(weights_name).has_shape()) {
|
| convert_pure_conv_to_depthwise.cc | 55 const auto& weights_name = conv_op->inputs[1]; local
56 if (CountOpsWithInput(*model, weights_name) > 1) {
64 auto& weights_array = model->GetArray(weights_name);
|
| shuffle_fc_weights.cc | 40 const string& weights_name = fc_op->inputs[1]; local
41 Array& weights_array = model->GetArray(weights_name);
109 if (CountOpsWithInput(*model, weights_name) != 1) {
|
| fuse_binary_into_preceding_affine.cc | 103 const auto& weights_name = preceding_op->inputs[1]; local
105 auto& weights = model->GetArray(weights_name);
106 DropMinMax(model, weights_name);
280 const auto& weights_name = preceding_op->inputs[1]; local
282 const auto& weights = model->GetArray(weights_name);
298 CountOpsWithInput(*model, weights_name);
|
| fuse_binary_into_following_affine.cc | 123 const auto& weights_name = following_op->inputs[1]; local
125 auto& weights = model->GetArray(weights_name);
126 DropMinMax(model, weights_name);
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/client/ |
| random_forest.py | 116 def _get_default_head(params, weights_name, output_type, name=None):
121 weight_column_name=weights_name,
128 weight_column_name=weights_name,
134 weight_column=weights_name,
141 weight_column=weights_name,
147 weight_column=weights_name,
155 weights_name=None,
169 model_head = _get_default_head(params, weights_name, output_type)
188 if weights_name and weights_name in features
[all...] |
| /external/tensorflow/tensorflow/lite/toco/ |
| import_tensorflow.cc | 629 const auto& weights_name = node.input(1); local
631 AvailableArrayName(*model, weights_name + "_reordered");
643 reorder->inputs = {weights_name};
706 const auto& weights_name = node.input(1); local
707 const auto& reordered_weights_name = weights_name + "_reordered";
719 reorder->inputs = {weights_name};
1884 const string& weights_name = node.input(TransposeConvOperator::WEIGHTS); local
[all...] |
| tooling_util.cc | 2363 const string& weights_name = fc_op.inputs[1]; local
[all...] |
| export_tensorflow.cc | 1480 const string weights_name = WalkUpToConstantArray( local
[all...] |
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| operator.cc | 492 const string& weights_name = op_signature.op->inputs[1]; variable
495 const Array& weights_array = op_signature.model->GetArray(weights_name);
1011 const string& weights_name = op_signature.op->inputs[2]; variable
1080 const string& weights_name = op_signature.op->inputs[2]; variable
[all...] |
