Searched refs:OptionalTensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.h | 39 struct OptionalTensor { struct
60 std::map<int, OptionalTensor>* result);
153 const std::map<int, OptionalTensor>& variable_args, OpKernelContext* ctx,
165 const std::map<int, OptionalTensor>& variables,
|
| D | xla_compile_on_demand_op.cc | 30 std::map<int, OptionalTensor> GetVariables(OpKernelContext* ctx) { in GetVariables()
31 std::map<int, OptionalTensor> variables; in GetVariables()
36 OptionalTensor& optional = variables[i]; in GetVariables()
54 std::map<int, OptionalTensor> variables = GetVariables(ctx); in Run()
200 std::map<int, OptionalTensor> variable_args = GetVariables(ctx); in Compile()
|
| D | xla_launch_util.cc | 149 std::map<int, OptionalTensor>* result) { in SnapshotResourceVariables()
157 OptionalTensor& tensor = (*result)[variable_indices[i]]; in SnapshotResourceVariables()
162 (*result)[variable_indices[i]] = OptionalTensor(); in SnapshotResourceVariables()
209 const std::map<int, OptionalTensor>& variables, in PopulateInputs()
445 const std::map<int, OptionalTensor>& variable_args, OpKernelContext* ctx, in BuildXlaCompilerArguments()
475 const OptionalTensor& variable = variable_args.at(input_num); in BuildXlaCompilerArguments()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 116 std::map<int, OptionalTensor> resource_var_snapshots, in XlaExecutableClosure()
132 const std::map<int, OptionalTensor>& resource_var_snapshots() const { in resource_var_snapshots()
141 std::map<int, OptionalTensor> resource_var_snapshots_;
258 std::map<int, OptionalTensor>* variables, in CompileToLocalExecutable()
330 std::map<int, OptionalTensor> variables; in Compute()
447 std::map<int, OptionalTensor> variables; in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | map_stage_op.cc | 85 using OptionalTensor = gtl::optional<Tensor>; typedef in tensorflow::__anonf2748bbc0111::StagingMap
86 using OptionalTuple = std::vector<OptionalTensor>;
158 [](const std::size_t& lhs, const OptionalTensor& rhs) { in get_tuple_bytes()
309 [](const OptionalTensor& v) { return v.has_value(); }); in put_incomplete()
415 [](const OptionalTensor& tensor) { return tensor.has_value(); })) { in pop()
450 [](const OptionalTensor& tensor) { return tensor.has_value(); })) { in popitem()
|