Home
last modified time | relevance | path

Searched defs:weights_array (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dcreate_im2col_arrays.cc33 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessConvOperator() local
Dconvert_pure_conv_to_depthwise.cc64 auto& weights_array = model->GetArray(weights_name); in Run() local
Dshuffle_fc_weights.cc41 Array& weights_array = model->GetArray(weights_name); in Run() local
Dpropagate_fixed_sizes.cc135 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessConvOperator() local
196 const auto& weights_array = in ProcessTransposeConvOperator() local
262 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessDepthwiseConvOperator() local
377 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessFullyConnectedOperator() local
1011 const auto& weights_array = in ProcessLstmCellOperator() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dfuse_binary_into_following_affine_test.cc103 auto& weights_array = model_->GetArray(op->inputs[1]); in TEST_F() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc400 const auto& weights_array = model.GetArray(weights_array_name); in ConvertConvOperator() local
553 const auto& weights_array = model.GetArray(weights_array_name); in ConvertTransposeConvOperator() local
1482 const auto& weights_array = model.GetArray(weights_name); in ConvertLstmCellOperator() local
Dtooling_util.cc1880 const auto& weights_array = model.GetArray(op.inputs[1]); in EstimateArithmeticOpsCount() local
2365 auto& weights_array = model->GetArray(weights_name); in UndoWeightsShuffling() local
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc495 const Array& weights_array = op_signature.model->GetArray(weights_name); in GetVersion() local
1014 const Array& weights_array = op_signature.model->GetArray(weights_name); in GetVersion() local
1083 const Array& weights_array = op_signature.model->GetArray(weights_name); in GetVersion() local