Searched refs:bias_array (Results 1 – 5 of 5) sorted by relevance
56 auto& bias_array = model->GetOrCreateArray(bias_name); in ProcessLinearOperator() local57 bias_array.data_type = ArrayDataType::kFloat; in ProcessLinearOperator()58 bias_array.mutable_shape()->mutable_dims()->push_back(depth); in ProcessLinearOperator()59 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in ProcessLinearOperator()
121 auto& bias_array = model->GetOrCreateArray(merged_biases); in Run() local122 bias_array.data_type = ArrayDataType::kFloat; in Run()123 bias_array.copy_shape(Shape({weights_dim1})); in Run()124 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run()
1020 const auto& bias_array = in ProcessLstmCellOperator() local1023 if (!bias_array.has_shape()) { in ProcessLstmCellOperator()1026 const auto& bias_shape = bias_array.shape(); in ProcessLstmCellOperator()1149 const auto& bias_array = model->GetArray(op->inputs[3]); in ProcessUnidirectionalSequenceRnnOperator() local1151 if (!bias_array.has_shape()) { in ProcessUnidirectionalSequenceRnnOperator()1155 const auto& bias_shape = bias_array.shape(); in ProcessUnidirectionalSequenceRnnOperator()1749 const auto& bias_array = model->GetArray(op->inputs[3]); in ProcessSvdfOperator() local1750 if (!bias_array.has_shape()) return; in ProcessSvdfOperator()
107 auto& bias_array = model_->GetArray(op->inputs[2]); in TEST_F() local108 EXPECT_THAT(bias_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
437 const auto& bias_array = model.GetArray(bias_array_name); in ConvertConvOperator() local439 Shape bias_shape_1d = bias_array.shape(); in ConvertConvOperator()441 CHECK(bias_array.buffer->type == ArrayDataType::kFloat); in ConvertConvOperator()443 bias_array.GetBuffer<ArrayDataType::kFloat>().data.data(); in ConvertConvOperator()527 const auto& bias_array = model.GetArray(bias_name); in ConvertDepthwiseConvOperator() local529 Shape bias_shape_1d = bias_array.shape(); in ConvertDepthwiseConvOperator()531 CHECK(bias_array.buffer->type == ArrayDataType::kFloat); in ConvertDepthwiseConvOperator()533 bias_array.GetBuffer<ArrayDataType::kFloat>().data.data(); in ConvertDepthwiseConvOperator()660 const auto& bias_array = model.GetArray(src_op.inputs[2]); in ConvertFullyConnectedOperator() local662 Shape bias_shape_1d = bias_array.shape(); in ConvertFullyConnectedOperator()[all …]