Home
last modified time | relevance | path

Searched refs:bias_array (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Densure_bias_vectors.cc68 auto& bias_array = model->GetOrCreateArray(bias_name); in ProcessLinearOperator() local
69 bias_array.data_type = ArrayDataType::kFloat; in ProcessLinearOperator()
70 bias_array.mutable_shape()->mutable_dims()->push_back(depth); in ProcessLinearOperator()
71 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in ProcessLinearOperator()
H A Didentify_lstm_merge_inputs.cc124 auto& bias_array = model->GetOrCreateArray(merged_biases); in Run() local
125 bias_array.data_type = ArrayDataType::kFloat; in Run()
126 bias_array.copy_shape(Shape({weights_dim1})); in Run()
127 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run()
H A Dpropagate_fixed_sizes.cc1033 const auto& bias_array = in ProcessLstmCellOperator() local
1036 if (!bias_array.has_shape()) { in ProcessLstmCellOperator()
1039 const auto& bias_shape = bias_array.shape(); in ProcessLstmCellOperator()
1162 const auto& bias_array = model->GetArray(op->inputs[3]); in ProcessUnidirectionalSequenceRnnOperator() local
1164 if (!bias_array.has_shape()) { in ProcessUnidirectionalSequenceRnnOperator()
1168 const auto& bias_shape = bias_array.shape(); in ProcessUnidirectionalSequenceRnnOperator()
1770 const auto& bias_array = model->GetArray(op->inputs[3]); in ProcessSvdfOperator() local
1771 if (!bias_array.has_shape()) return; in ProcessSvdfOperator()
/aosp_15_r20/external/mesa3d/src/gallium/targets/teflon/
H A Dtest_executor.cpp109 xt::xarray<int32_t> bias_array = xt::random::randint<int32_t>({output_channels}, -20000, 20000); in patch_conv2d() local
110 bias_data->resize(bias_array.size() * sizeof(int32_t)); in patch_conv2d()
111 memcpy(bias_data->data(), bias_array.data(), bias_array.size() * sizeof(int32_t)); in patch_conv2d()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dlstm_utils.cc710 Operation* bias_array; in ConvertKerasLSTMLayer() local
712 &bias_array))) in ConvertKerasLSTMLayer()
740 /*input_gate_bias=*/bias_array->getResult(0), in ConvertKerasLSTMLayer()
741 /*forget_gate_bias=*/bias_array->getResult(1), in ConvertKerasLSTMLayer()
742 /*cell_bias=*/bias_array->getResult(2), in ConvertKerasLSTMLayer()
743 /*output_gate_bias=*/bias_array->getResult(3), in ConvertKerasLSTMLayer()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dexport_tensorflow.cc446 const auto& bias_array = model.GetArray(bias_array_name); in ConvertConvOperator() local
448 Shape bias_shape_1d = bias_array.shape(); in ConvertConvOperator()
450 CHECK(bias_array.buffer->type == ArrayDataType::kFloat); in ConvertConvOperator()
452 bias_array.GetBuffer<ArrayDataType::kFloat>().data.data(); in ConvertConvOperator()
537 const auto& bias_array = model.GetArray(bias_name); in ConvertDepthwiseConvOperator() local
539 Shape bias_shape_1d = bias_array.shape(); in ConvertDepthwiseConvOperator()
541 CHECK(bias_array.buffer->type == ArrayDataType::kFloat); in ConvertDepthwiseConvOperator()
543 bias_array.GetBuffer<ArrayDataType::kFloat>().data.data(); in ConvertDepthwiseConvOperator()
670 const auto& bias_array = model.GetArray(src_op.inputs[2]); in ConvertFullyConnectedOperator() local
672 Shape bias_shape_1d = bias_array.shape(); in ConvertFullyConnectedOperator()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
H A Dfuse_binary_into_preceding_affine_test.cc113 auto& bias_array = model_->GetArray(op->inputs[3]); in TEST_F() local
114 EXPECT_THAT(bias_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
H A Dfuse_binary_into_following_affine_test.cc109 auto& bias_array = model_->GetArray(op->inputs[2]); in TEST_F() local
110 EXPECT_THAT(bias_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
/aosp_15_r20/external/mesa3d/src/amd/compiler/tests/
H A Dtest_d3d11_derivs.cpp258 BEGIN_TEST(d3d11_derivs.bias_array)