Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_batch_normalization.cc39 auto& mean_array = model->GetArray(bn_op->inputs[1]); in Run() local
46 if (!mean_array.buffer || !multiplier_array.buffer || !offset_array.buffer) { in Run()
58 CHECK(mean_array.data_type == ArrayDataType::kFloat); in Run()
95 auto& mean_shape = *mean_array.mutable_shape(); in Run()
100 auto& data = mean_array.GetMutableBuffer<ArrayDataType::kFloat>().data; in Run()
123 mean_array.GetBuffer<ArrayDataType::kFloat>().data; in Run()