Searched refs:bias_array (Results 1 – 7 of 7) sorted by relevance
68 auto& bias_array = model->GetOrCreateArray(bias_name); in ProcessLinearOperator() local69 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()
122 auto& bias_array = model->GetOrCreateArray(merged_biases); in Run() local123 bias_array.data_type = ArrayDataType::kFloat; in Run()124 bias_array.copy_shape(Shape({weights_dim1})); in Run()125 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run()
1033 const auto& bias_array = in ProcessLstmCellOperator() local1036 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() local1164 if (!bias_array.has_shape()) { in ProcessUnidirectionalSequenceRnnOperator()1168 const auto& bias_shape = bias_array.shape(); in ProcessUnidirectionalSequenceRnnOperator()1769 const auto& bias_array = model->GetArray(op->inputs[3]); in ProcessSvdfOperator() local1770 if (!bias_array.has_shape()) return; in ProcessSvdfOperator()
112 auto& bias_array = model_->GetArray(op->inputs[3]); in TEST_F() local113 EXPECT_THAT(bias_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
108 auto& bias_array = model_->GetArray(op->inputs[2]); in TEST_F() local109 EXPECT_THAT(bias_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
704 Operation* bias_array; in ConvertKerasLSTMLayer() local706 &bias_array))) in ConvertKerasLSTMLayer()735 /*input_gate_bias=*/bias_array->getResult(0), in ConvertKerasLSTMLayer()736 /*forget_gate_bias=*/bias_array->getResult(1), in ConvertKerasLSTMLayer()737 /*cell_bias=*/bias_array->getResult(2), in ConvertKerasLSTMLayer()738 /*output_gate_bias=*/bias_array->getResult(3), in ConvertKerasLSTMLayer()
441 const auto& bias_array = model.GetArray(bias_array_name); in ConvertConvOperator() local443 Shape bias_shape_1d = bias_array.shape(); in ConvertConvOperator()445 CHECK(bias_array.buffer->type == ArrayDataType::kFloat); in ConvertConvOperator()447 bias_array.GetBuffer<ArrayDataType::kFloat>().data.data(); in ConvertConvOperator()532 const auto& bias_array = model.GetArray(bias_name); in ConvertDepthwiseConvOperator() local534 Shape bias_shape_1d = bias_array.shape(); in ConvertDepthwiseConvOperator()536 CHECK(bias_array.buffer->type == ArrayDataType::kFloat); in ConvertDepthwiseConvOperator()538 bias_array.GetBuffer<ArrayDataType::kFloat>().data.data(); in ConvertDepthwiseConvOperator()665 const auto& bias_array = model.GetArray(src_op.inputs[2]); in ConvertFullyConnectedOperator() local667 Shape bias_shape_1d = bias_array.shape(); in ConvertFullyConnectedOperator()[all …]