Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc136 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessConvOperator() local
195 const auto& weights_array = in ProcessTransposeConvOperator() local
261 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
H A Dcreate_im2col_arrays.cc33 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessConvOperator() local
H A Dconvert_pure_conv_to_depthwise.cc64 auto& weights_array = model->GetArray(weights_name); in Run() local
H A Dshuffle_fc_weights.cc42 Array& weights_array = model->GetArray(weights_name); in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
H A Dfuse_binary_into_preceding_affine_test.cc109 auto& weights_array = model_->GetArray(op->inputs[1]); in TEST_F() local
H A Dfuse_binary_into_following_affine_test.cc105 auto& weights_array = model_->GetArray(op->inputs[1]); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dtooling_util.cc1906 const auto& weights_array = model.GetArray(op.inputs[1]); in EstimateArithmeticOpsCount() local
1925 const auto& weights_array = model.GetArray(op.inputs[1]); in EstimateArithmeticOpsCount() local
2434 auto& weights_array = model->GetArray(weights_name); in UndoWeightsShuffling() local
H A Dexport_tensorflow.cc409 const auto& weights_array = model.GetArray(weights_array_name); in ConvertConvOperator() local
563 const auto& weights_array = model.GetArray(weights_array_name); in ConvertTransposeConvOperator() local
1517 const auto& weights_array = model.GetArray(weights_name); in ConvertLstmCellOperator() local
/aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/
H A Dtest_executor.cpp136 xt::xarray<uint8_t> weights_array = xt::random::randint<uint8_t>(weight_shape, 0, 255); in patch_conv2d() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dlstm_utils.cc696 Operation* weights_array; in ConvertKerasLSTMLayer() local