Home
last modified time | relevance | path

Searched refs:weights_name (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest.py116 def _get_default_head(params, weights_name, output_type, name=None): argument
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, argument
169 model_head = _get_default_head(params, weights_name, output_type)
188 if weights_name and weights_name in features:
189 weights = features.pop(weights_name)
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Densure_bias_vectors.cc30 const string& weights_name = op.inputs[1]; in GetOutputDepthFromWeights() local
31 const auto& weights_shape = model.GetArray(weights_name).shape(); in GetOutputDepthFromWeights()
48 const string& weights_name = op->inputs[1]; in ProcessLinearOperator() local
49 if (!model->GetArray(weights_name).has_shape()) { in ProcessLinearOperator()
Dfuse_binary_into_preceding_affine.cc103 const auto& weights_name = preceding_op->inputs[1]; in FuseMulOrDivParamsIntoPrecedingAffine() local
105 auto& weights = model->GetArray(weights_name); in FuseMulOrDivParamsIntoPrecedingAffine()
106 DropMinMax(model, weights_name); in FuseMulOrDivParamsIntoPrecedingAffine()
280 const auto& weights_name = preceding_op->inputs[1]; in Run() local
282 const auto& weights = model->GetArray(weights_name); in Run()
298 CountOpsWithInput(*model, weights_name); in Run()
Dconvert_pure_conv_to_depthwise.cc55 const auto& weights_name = conv_op->inputs[1]; in Run() local
56 if (CountOpsWithInput(*model, weights_name) > 1) { in Run()
64 auto& weights_array = model->GetArray(weights_name); in Run()
Dshuffle_fc_weights.cc40 const string& weights_name = fc_op->inputs[1]; in Run() local
41 Array& weights_array = model->GetArray(weights_name); in Run()
109 if (CountOpsWithInput(*model, weights_name) != 1) { in Run()
Dfuse_binary_into_following_affine.cc123 const auto& weights_name = following_op->inputs[1]; in FuseMulOrDivParamsIntoFollowingAffine() local
125 auto& weights = model->GetArray(weights_name); in FuseMulOrDivParamsIntoFollowingAffine()
126 DropMinMax(model, weights_name); in FuseMulOrDivParamsIntoFollowingAffine()
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc629 const auto& weights_name = node.input(1); in ConvertConvOperator() local
631 AvailableArrayName(*model, weights_name + "_reordered"); in ConvertConvOperator()
643 reorder->inputs = {weights_name}; in ConvertConvOperator()
706 const auto& weights_name = node.input(1); in ConvertDepthwiseConvOperator() local
707 const auto& reordered_weights_name = weights_name + "_reordered"; in ConvertDepthwiseConvOperator()
719 reorder->inputs = {weights_name}; in ConvertDepthwiseConvOperator()
1884 const string& weights_name = node.input(TransposeConvOperator::WEIGHTS); in ConvertTransposeConvOperator() local
1885 const string& transposed_weights_name = weights_name + "_transposed"; in ConvertTransposeConvOperator()
1900 transpose->inputs = {weights_name, perm_array}; in ConvertTransposeConvOperator()
Dtooling_util.cc2363 const string& weights_name = fc_op.inputs[1]; in UndoWeightsShuffling() local
2364 QCHECK_EQ(CountOpsWithInput(*model, weights_name), 1); in UndoWeightsShuffling()
2365 auto& weights_array = model->GetArray(weights_name); in UndoWeightsShuffling()
Dexport_tensorflow.cc1480 const string weights_name = WalkUpToConstantArray( in ConvertLstmCellOperator() local
1482 const auto& weights_array = model.GetArray(weights_name); in ConvertLstmCellOperator()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc492 const string& weights_name = op_signature.op->inputs[1]; in GetVersion() local
495 const Array& weights_array = op_signature.model->GetArray(weights_name); in GetVersion()
1011 const string& weights_name = op_signature.op->inputs[2]; in GetVersion() local
1014 const Array& weights_array = op_signature.model->GetArray(weights_name); in GetVersion()
1080 const string& weights_name = op_signature.op->inputs[2]; in GetVersion() local
1083 const Array& weights_array = op_signature.model->GetArray(weights_name); in GetVersion()