Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dconvert_pure_conv_to_depthwise.cc64 auto& weights_array = model->GetArray(weights_name); in Run() local
65 if (!weights_array.buffer) { in Run()
69 if (weights_array.data_type != ArrayDataType::kFloat) { in Run()
93 depthwiseconv_op->depth_multiplier = weights_array.shape().dims(0); in Run()
98 const auto& weights_shape = weights_array.shape(); in Run()
100 weights_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run()
113 *weights_array.mutable_shape()->mutable_dims() = {1, width, height, depth}; in Run()
H A Dshuffle_fc_weights.cc42 Array& weights_array = model->GetArray(weights_name); in Run() local
48 weights_array.data_type != ArrayDataType::kUint8 || in Run()
50 !input_array.quantization_params || !weights_array.quantization_params || in Run()
55 if (!input_array.has_shape() || !weights_array.has_shape()) { in Run()
92 const Shape& weights_shape = weights_array.shape(); in Run()
106 if (!weights_array.buffer) { in Run()
119 weights_array.GetMutableBuffer<ArrayDataType::kUint8>().data; in Run()
H A Dcreate_im2col_arrays.cc33 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessConvOperator() local
34 if (!weights_array.has_shape()) { in ProcessConvOperator()
40 const auto& weights_shape = weights_array.shape(); in ProcessConvOperator()
H A Dpropagate_fixed_sizes.cc136 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessConvOperator() local
138 if (!weights_array.has_shape()) { in ProcessConvOperator()
141 const auto& weights_shape = weights_array.shape(); in ProcessConvOperator()
195 const auto& weights_array = in ProcessTransposeConvOperator() local
197 if (!weights_array.has_shape()) { in ProcessTransposeConvOperator()
201 const auto& weights_shape = weights_array.shape(); in ProcessTransposeConvOperator()
261 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessDepthwiseConvOperator() local
263 if (!weights_array.has_shape()) { in ProcessDepthwiseConvOperator()
266 const auto& weights_shape = weights_array.shape(); in ProcessDepthwiseConvOperator()
378 const auto& weights_array = model->GetArray(op->inputs[1]); in ProcessFullyConnectedOperator() local
[all …]
/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
137 weights_data->resize(weights_array.size()); in patch_conv2d()
138 memcpy(weights_data->data(), weights_array.data(), weights_array.size()); in patch_conv2d()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dlstm_utils.cc696 Operation* weights_array; in ConvertKerasLSTMLayer() local
699 &weights_array))) in ConvertKerasLSTMLayer()
729 /*input_to_input_weights=*/weights_array->getResult(0), in ConvertKerasLSTMLayer()
730 /*input_to_forget_weights=*/weights_array->getResult(1), in ConvertKerasLSTMLayer()
731 /*input_to_cell_weights=*/weights_array->getResult(2), in ConvertKerasLSTMLayer()
732 /*input_to_output_weights=*/weights_array->getResult(3), in ConvertKerasLSTMLayer()
/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
110 EXPECT_THAT(weights_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
H A Dfuse_binary_into_following_affine_test.cc105 auto& weights_array = model_->GetArray(op->inputs[1]); in TEST_F() local
106 EXPECT_THAT(weights_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
/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
1907 if (!output_array.has_shape() || !weights_array.has_shape()) { in EstimateArithmeticOpsCount()
1915 2 * RequiredBufferSizeForShape(weights_array.shape()); in EstimateArithmeticOpsCount()
1925 const auto& weights_array = model.GetArray(op.inputs[1]); in EstimateArithmeticOpsCount() local
1926 if (!input_array.has_shape() || !weights_array.has_shape()) { in EstimateArithmeticOpsCount()
1930 const Shape& weights = weights_array.shape(); in EstimateArithmeticOpsCount()
2434 auto& weights_array = model->GetArray(weights_name); in UndoWeightsShuffling() local
2435 QCHECK(weights_array.data_type == ArrayDataType::kUint8); in UndoWeightsShuffling()
2437 weights_array.GetMutableBuffer<toco::ArrayDataType::kUint8>().data; in UndoWeightsShuffling()
2438 const auto& weights_shape = weights_array.shape(); in UndoWeightsShuffling()
H A Dexport_tensorflow.cc409 const auto& weights_array = model.GetArray(weights_array_name); in ConvertConvOperator() local
410 CHECK(weights_array.buffer->type == ArrayDataType::kFloat); in ConvertConvOperator()
563 const auto& weights_array = model.GetArray(weights_array_name); in ConvertTransposeConvOperator() local
564 CHECK(weights_array.buffer->type == ArrayDataType::kFloat); in ConvertTransposeConvOperator()
1517 const auto& weights_array = model.GetArray(weights_name); in ConvertLstmCellOperator() local
1519 const auto& weights_shape = weights_array.shape(); in ConvertLstmCellOperator()
1521 CHECK(weights_array.buffer); in ConvertLstmCellOperator()
1522 CHECK(weights_array.buffer->type == ArrayDataType::kFloat); in ConvertLstmCellOperator()
1524 weights_array.GetBuffer<ArrayDataType::kFloat>().data.data(); in ConvertLstmCellOperator()