HomeSort by relevance Sort by last modified time
    Searched refs:bias_array (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
ensure_bias_vectors.cc 56 auto& bias_array = model->GetOrCreateArray(bias_name); local
57 bias_array.data_type = ArrayDataType::kFloat;
58 bias_array.mutable_shape()->mutable_dims()->push_back(depth);
59 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>();
identify_lstm_merge_inputs.cc 121 auto& bias_array = model->GetOrCreateArray(merged_biases); local
122 bias_array.data_type = ArrayDataType::kFloat;
123 bias_array.copy_shape(Shape({weights_dim1}));
124 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>();
propagate_fixed_sizes.cc 1020 const auto& bias_array = local
1149 const auto& bias_array = model->GetArray(op->inputs[3]); local
1749 const auto& bias_array = model->GetArray(op->inputs[3]); local
    [all...]
  /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
fuse_binary_into_following_affine_test.cc 107 auto& bias_array = model_->GetArray(op->inputs[2]); local
108 EXPECT_THAT(bias_array.GetBuffer<toco::ArrayDataType::kFloat>().data,
  /external/tensorflow/tensorflow/lite/toco/
export_tensorflow.cc 437 const auto& bias_array = model.GetArray(bias_array_name); local
439 Shape bias_shape_1d = bias_array.shape();
441 CHECK(bias_array.buffer->type == ArrayDataType::kFloat);
443 bias_array.GetBuffer<ArrayDataType::kFloat>().data.data();
527 const auto& bias_array = model.GetArray(bias_name); local
529 Shape bias_shape_1d = bias_array.shape();
531 CHECK(bias_array.buffer->type == ArrayDataType::kFloat);
533 bias_array.GetBuffer<ArrayDataType::kFloat>().data.data();
660 const auto& bias_array = model.GetArray(src_op.inputs[2]); local
662 Shape bias_shape_1d = bias_array.shape()
1509 const auto& bias_array = model.GetArray(bias_name); local
    [all...]

Completed in 62 milliseconds