| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | variant_tensor_data.cc | 23 VariantTensorData::VariantTensorData(VariantTensorDataProto proto) { in VariantTensorData() function in tensorflow::VariantTensorData
27 int VariantTensorData::tensors_size() const { return tensors_.size(); } in tensors_size()
29 const Tensor& VariantTensorData::tensors(int index) const { in tensors()
33 const std::vector<Tensor>& VariantTensorData::tensors() const { in tensors()
37 Tensor* VariantTensorData::add_tensors() { in add_tensors()
42 void VariantTensorData::ToProto(VariantTensorDataProto* proto) const { in ToProto()
51 bool VariantTensorData::FromProto(VariantTensorDataProto proto) { in FromProto()
63 bool VariantTensorData::FromConstProto(const VariantTensorDataProto& proto) { in FromConstProto()
74 string VariantTensorData::SerializeAsString() const { in SerializeAsString()
80 bool VariantTensorData::SerializeToString(string* buf) { in SerializeToString()
[all …]
|
| H A D | variant_encode_decode.h | 52 VariantTensorData* data) { in EncodeVariantImpl()
60 VariantTensorData* data) { in EncodeVariantImpl()
70 VariantTensorData* data) { in EncodeVariantImpl()
79 VariantTensorData* data) { in EncodeVariantImpl()
85 bool DecodeVariantImpl(VariantTensorData data, in DecodeVariantImpl()
94 bool DecodeVariantImpl(VariantTensorData data, in DecodeVariantImpl()
104 bool DecodeVariantImpl(VariantTensorData data, in DecodeVariantImpl()
115 bool DecodeVariantImpl(VariantTensorData data, in DecodeVariantImpl()
222 void EncodeVariant(const T& value, VariantTensorData* data) {
228 bool DecodeVariant(VariantTensorData* data, T* value) {
[all …]
|
| H A D | variant_test.cc | 91 void Encode(VariantTensorData* data) const {} in Encode()
92 bool Decode(VariantTensorData data) { return false; } in Decode()
131 void Encode(VariantTensorData* data) const {} in Encode()
132 bool Decode(VariantTensorData data) { return false; } in Decode()
252 void Encode(VariantTensorData* data) const {} in Encode()
253 bool Decode(VariantTensorData data) { return false; } in Decode()
534 void Encode(VariantTensorData* data) const { data->tensors_ = vec; } in Encode()
536 bool Decode(VariantTensorData data) { in Decode()
574 VariantTensorData serialized; in TEST()
649 VariantTensorData serialized; in TestEncodeDecodePod()
[all …]
|
| H A D | variant_tensor_data.h | 38 class VariantTensorData {
40 VariantTensorData() = default;
44 VariantTensorData(VariantTensorDataProto proto);
126 std::string ProtoDebugString(const VariantTensorData& object);
129 Tensor* VariantTensorData::add_tensor(TensorConstructorArgs&&... args) { in add_tensor()
|
| H A D | variant.h | 43 bool DecodeVariant(VariantTensorData* data, T* value);
49 void EncodeVariant(const T& value, VariantTensorData* data);
278 void Encode(VariantTensorData* data) const { in Encode()
285 bool Decode(VariantTensorData data);
321 virtual void Encode(VariantTensorData* data) const = 0;
322 virtual bool Decode(VariantTensorData data) = 0;
369 void Encode(VariantTensorData* data) const final { in Encode()
373 bool Decode(VariantTensorData data) final { in Decode()
|
| H A D | variant.cc | 26 bool Variant::Decode(VariantTensorData data) { in Decode()
56 VariantTensorData* data) { in EncodeVariant()
61 bool DecodeVariant(VariantTensorData* data, VariantTensorDataProto* value) { in DecodeVariant()
|
| H A D | variant_op_copy_test.cc | 63 void Encode(VariantTensorData* data) const { data->tensors_ = {stored}; } in Encode()
64 bool Decode(const VariantTensorData& data) { in Decode()
171 VariantTensorData data; in TEST()
205 VariantTensorData data; in TEST()
248 VariantTensorData data; in TEST()
|
| H A D | shared_ptr_variant.h | 64 void Encode(VariantTensorData*) const { in Encode()
68 bool Decode(const VariantTensorData&) { in Decode()
|
| H A D | variant_op_registry_test.cc | 117 VariantTensorData data; in TEST()
134 VariantTensorData data; in TEST()
|
| H A D | dataset.cc | 105 void Encode(VariantTensorData* data) const { in Encode()
109 bool Decode(const VariantTensorData& data) { in Decode()
137 void Encode(VariantTensorData* data) const { in Encode()
141 bool Decode(const VariantTensorData& data) { in Decode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | serialization_utils_test.cc | 105 std::vector<const VariantTensorData*> data; in TEST()
134 std::vector<const VariantTensorData*> data; in TEST()
148 VariantTensorData data; in TEST()
151 std::vector<const VariantTensorData*> reader_data; in TEST()
171 std::vector<const VariantTensorData*> data; in TEST()
185 VariantTensorData data; in TEST()
188 std::vector<const VariantTensorData*> reader_data; in TEST()
207 std::vector<const VariantTensorData*> data; in TEST()
|
| H A D | serialization_utils.h | 45 const std::vector<const VariantTensorData*>& data);
73 std::map<string, const VariantTensorData*> data_; // Not owned.
102 void ReleaseData(std::vector<std::unique_ptr<VariantTensorData>>* variants);
105 void GetData(std::vector<const VariantTensorData*>* variants);
119 std::map<string, std::unique_ptr<VariantTensorData>> data_;
|
| H A D | serialization_utils.cc | 151 const std::vector<const tensorflow::VariantTensorData*>& data) { in VariantTensorDataReader()
345 std::vector<std::unique_ptr<VariantTensorData>>* variants) { in ReleaseData()
354 std::vector<const VariantTensorData*>* variants) { in GetData()
392 data_[name] = std::make_unique<VariantTensorData>(); in WriteTensorInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | iterator_ops.cc | 301 Status InitializeFromVariantData(std::unique_ptr<VariantTensorData> d) { in InitializeFromVariantData()
307 void Encode(VariantTensorData* data) const { *data = *data_; } in Encode()
308 bool Decode(VariantTensorData data) { in Decode()
312 auto tensor_data = std::make_unique<VariantTensorData>(); in Decode()
319 const VariantTensorData* GetData() const { return data_.get(); } in GetData()
331 std::unique_ptr<VariantTensorData> data_;
366 std::vector<std::unique_ptr<VariantTensorData>> data; in InitializeFromIterator()
384 std::vector<const VariantTensorData*> data; in InitFromTensor()
|
| H A D | optional_ops_util.h | 59 void Encode(VariantTensorData* data) const { in Encode()
68 bool Decode(const VariantTensorData& data) { in Decode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | composite_tensor_variant.h | 70 void Encode(VariantTensorData* data) const;
73 bool Decode(const VariantTensorData& data);
|
| H A D | composite_tensor_variant.cc | 45 void CompositeTensorVariant::Encode(VariantTensorData* data) const { in Encode()
53 bool CompositeTensorVariant::Decode(const VariantTensorData& data) { in Decode()
|
| H A D | ragged_tensor_variant.cc | 35 void RaggedTensorVariant::Encode(VariantTensorData* data) const { in Encode()
43 bool RaggedTensorVariant::Decode(const VariantTensorData& data) { in Decode()
|
| H A D | tensor_map.cc | 29 void TensorMap::Encode(VariantTensorData* data) const { in Encode()
69 bool TensorMap::Decode(const VariantTensorData& data) { in Decode()
|
| H A D | tensor_list.h | 110 void Encode(VariantTensorData* data) const;
112 bool Decode(const VariantTensorData& data);
|
| H A D | tensor_cord.cc | 35 void TensorCord::Encode(VariantTensorData* data) const { in Encode()
42 bool TensorCord::Decode(VariantTensorData data) { in Decode()
|
| H A D | ragged_tensor_variant.h | 46 void Encode(VariantTensorData* data) const;
47 bool Decode(const VariantTensorData& data);
|
| H A D | tensor_list.cc | 28 void TensorList::Encode(VariantTensorData* data) const { in Encode()
54 bool TensorList::Decode(const VariantTensorData& data) { in Decode()
|
| H A D | tensor_map.h | 98 void Encode(VariantTensorData* data) const;
100 bool Decode(const VariantTensorData& data);
|
| H A D | tensor_cord.h | 197 void Encode(VariantTensorData* data) const;
199 bool Decode(VariantTensorData data);
|