OpenGrok
| Home | Sort by relevance Sort by last modified time |
Searched refs:tensor_store (Results 1 -
6 of 6) sorted by null
Completed in 90 milliseconds
| /external/tensorflow/tensorflow/core/common_runtime/ | |
| executor.h | 90 TensorStore* tensor_store = nullptr; member in struct:tensorflow::Executor::Args |
| direct_session.cc | 530 args.tensor_store = &run_state.tensor_store; 672 run_state.tensor_store.SaveTensors(output_names, &session_state_)); 772 args.tensor_store = &run_state->tensor_store; 857 s = run_state->tensor_store.SaveTensors(output_names, &session_state_); [all...] |
| direct_session.h | 172 TensorStore tensor_store; member in struct:tensorflow::DirectSession::RunState |
| executor.cc | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ | |
| session_ops.cc | 47 OP_REQUIRES_OK(ctx, ctx->tensor_store()->AddTensor(name(), tk)); |
| /external/tensorflow/tensorflow/core/framework/ | |
| op_kernel.h | 539 TensorStore* tensor_store = nullptr; member in struct:tensorflow::OpKernelContext::Params 941 TensorStore* tensor_store() const { return params_->tensor_store; } function in class:tensorflow::OpKernelContext [all...] |
Completed in 90 milliseconds
