Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsave_op.cc34 SaveTensors(context, &checkpoint::CreateTableTensorSliceBuilder, false); in Compute()
45 SaveTensors(context, &checkpoint::CreateTableTensorSliceBuilder, true); in Compute()
Dsave_restore_tensor.h35 void SaveTensors(
Dsave_restore_tensor.cc40 void SaveTensors( in SaveTensors() function
/external/tensorflow/tensorflow/core/framework/
Dsession_state.h74 Status SaveTensors(const std::vector<string>& output_names,
/external/tensorflow/tensorflow/core/common_runtime/
Dsession_state.cc70 Status TensorStore::SaveTensors(const std::vector<string>& output_names, in SaveTensors() function in tensorflow::TensorStore
Ddirect_session.cc766 TF_RETURN_IF_ERROR(run_state.tensor_store.SaveTensors( in RunInternal()
1085 s = run_state->tensor_store.SaveTensors(output_names, &session_state_); in PRun()