Home
last modified time | relevance | path

Searched refs:VariantTensorData (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dvariant_tensor_data.cc23 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 …]
Dvariant_encode_decode.h52 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 …]
Dvariant_tensor_data.h38 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()
Dvariant.cc26 bool Variant::Decode(VariantTensorData data) { in Decode()
56 VariantTensorData* data) { in EncodeVariant()
61 bool DecodeVariant(VariantTensorData* data, VariantTensorDataProto* value) { in DecodeVariant()
Dvariant_test.cc91 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 …]
Dextension_type_variant.h73 void Encode(VariantTensorData* data) const;
76 bool Decode(const VariantTensorData& data);
Dvariant.h43 bool DecodeVariant(VariantTensorData* data, T* value);
49 void EncodeVariant(const T& value, VariantTensorData* data);
275 void Encode(VariantTensorData* data) const { in Encode()
282 bool Decode(VariantTensorData data);
318 virtual void Encode(VariantTensorData* data) const = 0;
319 virtual bool Decode(VariantTensorData data) = 0;
366 void Encode(VariantTensorData* data) const final { in Encode()
370 bool Decode(VariantTensorData data) final { in Decode()
Dextension_type_variant.cc25 void ExtensionTypeVariant::Encode(VariantTensorData* data) const { in Encode()
33 bool ExtensionTypeVariant::Decode(const VariantTensorData& data) { in Decode()
Dshared_ptr_variant.h63 void Encode(VariantTensorData*) const { in Encode()
67 bool Decode(const VariantTensorData&) { in Decode()
Dvariant_op_copy_test.cc63 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()
Ddataset.cc88 void Encode(VariantTensorData* data) const { in Encode()
92 bool Decode(const VariantTensorData& data) { in Decode()
120 void Encode(VariantTensorData* data) const { in Encode()
124 bool Decode(const VariantTensorData& data) { in Decode()
Dvariant_op_registry_test.cc117 VariantTensorData data; in TEST()
134 VariantTensorData data; in TEST()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils_test.cc53 std::vector<const VariantTensorData*> data; in TEST()
67 VariantTensorData data; in TEST()
70 std::vector<const VariantTensorData*> reader_data; in TEST()
90 std::vector<const VariantTensorData*> data; in TEST()
104 VariantTensorData data; in TEST()
107 std::vector<const VariantTensorData*> reader_data; in TEST()
126 std::vector<const VariantTensorData*> data; in TEST()
141 std::vector<const VariantTensorData*> data; in TEST()
Ddataset_utils.h166 const std::vector<const VariantTensorData*>& data);
192 std::map<string, const VariantTensorData*> data_; // Not owned.
220 void ReleaseData(std::vector<std::unique_ptr<VariantTensorData>>* variants);
223 void GetData(std::vector<const VariantTensorData*>* variants);
239 std::map<string, std::unique_ptr<VariantTensorData>> data_;
Doptional_ops.h65 void Encode(VariantTensorData* data) const { in Encode()
74 bool Decode(const VariantTensorData& data) { in Decode()
Diterator_ops.cc309 Status InitializeFromVariantData(std::unique_ptr<VariantTensorData> d) { in InitializeFromVariantData()
315 void Encode(VariantTensorData* data) const { *data = *data_; } in Encode()
316 bool Decode(VariantTensorData data) { in Decode()
320 auto tensor_data = absl::make_unique<VariantTensorData>(); in Decode()
327 const VariantTensorData* GetData() const { return data_.get(); } in GetData()
339 std::unique_ptr<VariantTensorData> data_;
374 std::vector<std::unique_ptr<VariantTensorData>> data; in InitializeFromIterator()
392 std::vector<const VariantTensorData*> data; in InitFromTensor()
Dsplit_utils_test.cc33 std::vector<const VariantTensorData*> variants; in SaveAndRestore()
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_variant.cc35 void RaggedTensorVariant::Encode(VariantTensorData* data) const { in Encode()
43 bool RaggedTensorVariant::Decode(const VariantTensorData& data) { in Decode()
Dtensor_map.cc29 void TensorMap::Encode(VariantTensorData* data) const { in Encode()
69 bool TensorMap::Decode(const VariantTensorData& data) { in Decode()
Dtensor_cord.cc35 void TensorCord::Encode(VariantTensorData* data) const { in Encode()
42 bool TensorCord::Decode(VariantTensorData data) { in Decode()
Dtensor_list.h110 void Encode(VariantTensorData* data) const;
112 bool Decode(const VariantTensorData& data);
Dragged_tensor_variant.h46 void Encode(VariantTensorData* data) const;
47 bool Decode(const VariantTensorData& data);
Dtensor_map.h98 void Encode(VariantTensorData* data) const;
100 bool Decode(const VariantTensorData& data);
Dtensor_list.cc28 void TensorList::Encode(VariantTensorData* data) const { in Encode()
80 bool TensorList::Decode(const VariantTensorData& data) { in Decode()
Dtensor_cord.h197 void Encode(VariantTensorData* data) const;
199 bool Decode(VariantTensorData data);

12