MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / ConvertFloatTensorConst

Function ConvertFloatTensorConst

tensorflow/lite/toco/export_tensorflow.cc:150–166  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

148}
149
150void ConvertFloatTensorConst(const string& name, const Shape& input_shape,
151 const float* input_data,
152 AxesOrder input_axes_order,
153 AxesOrder output_axes_order,
154 GraphDef* tensorflow_graph,
155 LegacyScalarPolicy legacy_scalar_policy) {
156 if (HasAlreadyExportedConst(name, *tensorflow_graph)) {
157 return;
158 }
159 tensorflow::NodeDef* const_op = tensorflow_graph->add_node();
160 const_op->set_op("Const");
161 const_op->set_name(name);
162 (*const_op->mutable_attr())["dtype"].set_type(DT_FLOAT);
163 auto* tensor = (*const_op->mutable_attr())["value"].mutable_tensor();
164 ExportFloatArray(input_axes_order, input_shape, input_data, output_axes_order,
165 tensor, legacy_scalar_policy);
166}
167
168void ConvertFloatTensorConst(const string& name, const Shape& input_shape,
169 const float* input_data,

Calls 7

HasAlreadyExportedConstFunction · 0.85
ExportFloatArrayFunction · 0.85
set_opMethod · 0.80
HasArrayMethod · 0.80
set_nameMethod · 0.45
shapeMethod · 0.45
dataMethod · 0.45

Tested by

no test coverage detected