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

Function TF_TryEvaluateConstant

tensorflow/c/c_api.cc:2379–2395  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

2377}
2378
2379unsigned char TF_TryEvaluateConstant(TF_Graph* graph, TF_Output output,
2380 TF_Tensor** result, TF_Status* status) {
2381 *result = nullptr;
2382 mutex_lock l(graph->mu);
2383 OutputTensor tensor(&output.oper->node, output.index);
2384 bool evaluated;
2385 Tensor result_tensor;
2386 status->status = EvaluateConstantTensor(
2387 tensor, graph->refiner, *graph->graph.op_registry(),
2388 graph->graph.versions().producer(), &evaluated, &result_tensor);
2389 if (evaluated) {
2390 DCHECK(TF_GetCode(status) == TF_OK);
2391 *result = TF_TensorFromTensor(result_tensor, status);
2392 if (TF_GetCode(status) != TF_OK) evaluated = false;
2393 }
2394 return evaluated;
2395}
2396
2397TF_ApiDefMap* TF_NewApiDefMap(TF_Buffer* op_list_buffer, TF_Status* status) {
2398 tensorflow::OpList op_list;

Callers 1

Calls 5

EvaluateConstantTensorFunction · 0.85
TF_GetCodeFunction · 0.85
TF_TensorFromTensorFunction · 0.85
op_registryMethod · 0.45
versionsMethod · 0.45

Tested by

no test coverage detected