Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api_internal.h147 std::map<tensorflow::TensorHandle*, TF_Output> input_tensor_map; member
155 for (auto input : input_tensor_map) { in ~TFE_TraceContext()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc9024 auto find_it = trace_ctx->input_tensor_map.find(handle); in getOrCreateSymbolicTensor()
9025 if (find_it != trace_ctx->input_tensor_map.end()) { in getOrCreateSymbolicTensor()
9046 trace_ctx->input_tensor_map[handle] = ret; in getOrCreateSymbolicTensor()
9121 trace_ctx->input_tensors->reserve(trace_ctx->input_tensor_map.size()); in TFE_FinalizeInputTensorsFromTraceContext()
9123 for (auto input : trace_ctx->input_tensor_map) { in TFE_FinalizeInputTensorsFromTraceContext()
9127 return trace_ctx->input_tensor_map.size(); in TFE_FinalizeInputTensorsFromTraceContext()