Searched refs:tensor_store (Results 1 – 7 of 7) sorted by relevance
211 params.tensor_store = args.tensor_store; in RunAsync()
93 TensorStore* tensor_store = nullptr; member
537 args.tensor_store = &run_state.tensor_store; in RunInternal()680 if (!run_state.tensor_store.empty()) { in RunInternal()681 TF_RETURN_IF_ERROR(run_state.tensor_store.SaveTensors( in RunInternal()893 args.tensor_store = &run_state->tensor_store; in PRunSetup()978 s = run_state->tensor_store.SaveTensors(output_names, &session_state_); in PRun()
189 TensorStore tensor_store; member
1386 tensor_store_(args.tensor_store), in ExecutorState()1632 params.tensor_store = tensor_store_; in Process()
47 OP_REQUIRES_OK(ctx, ctx->tensor_store()->AddTensor(name(), tk)); in Compute()
644 TensorStore* tensor_store = nullptr; member1079 TensorStore* tensor_store() const { return params_->tensor_store; } in tensor_store() function