Searched refs:packed_args_ (Results 1 – 3 of 3) sorted by relevance
40 packed_args_.emplace(index, gtl::InlinedVector<TensorValue, 4>(num_handles)); in InitPackedHandle()41 TensorValue* packed_arg_flat = &(packed_args_[index][0]); in InitPackedHandle()130 if (packed_args_.contains(index.index)) { in GetLocalArg()131 Tensor* arg = packed_args_.at(index.index).at(index.sub_index).tensor; in GetLocalArg()
85 absl::flat_hash_map<int, gtl::InlinedVector<TensorValue, 4>> packed_args_; variable
818 packed_args_.emplace(index, std::move(tensor_args)); in TestFunctionPackedArgs()827 *val = *packed_args_.at(index.index).at(index.sub_index).tensor; in GetLocalArg()834 absl::flat_hash_map<int, gtl::InlinedVector<TensorValue, 4>> packed_args_; member in tensorflow::__anon96a4e55d0111::TestFunctionPackedArgs