Home
last modified time | relevance | path

Searched defs:weights_array (Results 1 – 9 of 9) 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.cc136 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessConvOperator() local
197 const auto& weights_array = in ProcessTransposeConvOperator() local
263 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessDepthwiseConvOperator() local
378 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessFullyConnectedOperator() local
1024 const auto& weights_array = in ProcessLstmCellOperator() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dfuse_binary_into_preceding_affine_test.cc108 auto& weights_array = model_->GetArray(op->inputs[1]); in TEST_F() local
Dfuse_binary_into_following_affine_test.cc104 auto& weights_array = model_->GetArray(op->inputs[1]); in TEST_F() local
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc1904 const auto& weights_array = model.GetArray(op.inputs[1]); in EstimateArithmeticOpsCount() local
1923 const auto& weights_array = model.GetArray(op.inputs[1]); in EstimateArithmeticOpsCount() local
2427 auto& weights_array = model->GetArray(weights_name); in UndoWeightsShuffling() local
Dexport_tensorflow.cc404 const auto& weights_array = model.GetArray(weights_array_name); in ConvertConvOperator() local
558 const auto& weights_array = model.GetArray(weights_array_name); in ConvertTransposeConvOperator() local
1512 const auto& weights_array = model.GetArray(weights_name); in ConvertLstmCellOperator() local
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc690 Operation* weights_array; in ConvertKerasLSTMLayer() local