Searched refs:snapshot_ptrs (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 63 std::map<int, const Tensor*> snapshot_ptrs; in Run() local
65 snapshot_ptrs.emplace(p.first, in Run()
72 launch_context.PopulateInputs(ctx, result, snapshot_ptrs, in Run()
98 input_output_alias, snapshot_ptrs)); in Run()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 486 std::map<int, const Tensor*> snapshot_ptrs; in Compute() local
497 snapshot_ptrs.emplace(p.first, in Compute()
501 ctx, closure.compilation_result(), snapshot_ptrs, in Compute()
544 absl::MakeSpan(*variable_infos), input_output_alias, snapshot_ptrs)); in Compute()
|