HomeSort by relevance Sort by last modified time
    Searched full:weights_name (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/contrib/tensor_forest/client/
random_forest.py 109 def get_default_head(params, weights_name, name=None):
112 weight_column_name=weights_name,
119 weight_column_name=weights_name,
128 weights_name=None,
141 model_head = get_default_head(params, weights_name)
153 if weights_name and weights_name in features:
154 weights = features.pop(weights_name)
197 features[weights_name] = weights
378 weights_name=weight_column
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
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);
fuse_binary_into_preceding_affine.cc 91 const auto& weights_name = preceding_op->inputs[1]; local
93 auto& weights = model->GetArray(weights_name);
94 DropMinMax(model, weights_name);
propagate_fixed_sizes.cc 106 const string& weights_name = op.inputs[1];
107 const auto& weights_shape = model.GetArray(weights_name).shape();
119 const string& weights_name = op->inputs[1];
120 const auto& weights_array = model->GetArray(weights_name);
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
import_tensorflow.cc 374 const auto& weights_name = node.input(1); local
375 const auto& reordered_weights_name = weights_name + "_reordered";
387 reorder->inputs = {weights_name};
427 const auto& weights_name = node.input(1); local
428 const auto& reordered_weights_name = weights_name + "_reordered";
440 reorder->inputs = {weights_name};
    [all...]
export_tensorflow.cc 1239 const string weights_name = WalkUpToConstantArray( local
    [all...]

Completed in 117 milliseconds