Home
last modified time | relevance | path

Searched refs:VariantTensorData (Results 1 – 25 of 33) 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()
43 Tensor* VariantTensorData::add_tensor(TensorConstructorArgs&&... args) { in add_tensor()
48 void VariantTensorData::ToProto(VariantTensorDataProto* proto) const { in ToProto()
57 bool VariantTensorData::FromProto(VariantTensorDataProto proto) { in FromProto()
69 bool VariantTensorData::FromConstProto(const VariantTensorDataProto& proto) { in FromConstProto()
80 string VariantTensorData::SerializeAsString() const { in SerializeAsString()
[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.h43 bool DecodeVariant(VariantTensorData* data, T* value);
49 void EncodeVariant(const T& value, VariantTensorData* data);
234 void Encode(VariantTensorData* data) const { in Encode()
241 bool Decode(VariantTensorData data);
268 virtual void Encode(VariantTensorData* data) const = 0;
269 virtual bool Decode(VariantTensorData data) = 0;
298 void Encode(VariantTensorData* data) const override { in Encode()
302 bool Decode(VariantTensorData data) override { in Decode()
Dvariant_tensor_data.h38 class VariantTensorData {
40 VariantTensorData() = default;
44 VariantTensorData(VariantTensorDataProto proto);
120 string ProtoDebugString(const VariantTensorData& object);
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.cc145 void Encode(VariantTensorData* data) const { data->tensors_ = vec; } in Encode()
147 bool Decode(VariantTensorData data) { in Decode()
185 VariantTensorData serialized; in TEST()
247 VariantTensorData serialized; in TEST()
264 VariantTensorData serialized; in TEST()
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.cc65 void Encode(VariantTensorData* data) const { in Encode()
69 bool Decode(const VariantTensorData& data) { in Decode()
97 void Encode(VariantTensorData* data) const { in Encode()
101 bool Decode(const VariantTensorData& data) { in Decode()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.h63 explicit VariantTensorDataReader(const VariantTensorData* data);
77 const VariantTensorData* data_; // Not owned.
84 explicit VariantTensorDataWriter(VariantTensorData* data) : data_(data) {} in VariantTensorDataWriter()
97 VariantTensorData* data_;
Ddataset_utils_test.cc49 VariantTensorData data; in TEST()
68 VariantTensorData data; in TEST()
Doptional_ops.h65 void Encode(VariantTensorData* data) const { in Encode()
74 bool Decode(const VariantTensorData& data) { in Decode()
Diterator_ops.cc343 data_ = absl::make_unique<VariantTensorData>(); in InitializeFromIterator()
351 void Encode(VariantTensorData* data) const { *data = *data_; } in Encode()
352 bool Decode(VariantTensorData data) { in Decode()
356 std::unique_ptr<VariantTensorData> tensor_data = in Decode()
357 absl::make_unique<VariantTensorData>(); in Decode()
377 std::unique_ptr<VariantTensorData> data_;
Dconcatenate_dataset_op_test.cc269 VariantTensorData data; in TEST_F()
319 VariantTensorData data; in TEST_P()
Dzip_dataset_op_test.cc253 VariantTensorData data; in TEST_F()
318 VariantTensorData data; in TEST_P()
Drange_dataset_op_test.cc252 VariantTensorData data; in TEST_F()
397 VariantTensorData data; in TEST_P()
Dsparse_tensor_slice_dataset_op_test.cc321 VariantTensorData data; in TEST_F()
498 VariantTensorData data; in TEST_F()
Dtensor_dataset_op_test.cc344 VariantTensorData data; in TEST_P()
490 VariantTensorData data; in TEST_P()
Dtensor_slice_dataset_op_test.cc380 VariantTensorData data; in TEST_F()
598 VariantTensorData data; in TEST_F()
Drepeat_dataset_op_test.cc351 VariantTensorData data; in TEST_F()
516 VariantTensorData data; in TEST_P()
Dmap_dataset_op_test.cc334 VariantTensorData data; in TEST_F()
509 VariantTensorData data; in TEST_P()
Dtake_dataset_op_test.cc358 VariantTensorData data; in TEST_F()
524 VariantTensorData data; in TEST_P()
/external/tensorflow/tensorflow/core/kernels/
Dstring_view_variant_wrapper.h49 void Encode(VariantTensorData* data) const { in Encode()
55 bool Decode(const VariantTensorData& data) { in Decode()
Dmutex_ops.cc100 void Encode(VariantTensorData*) const { in Encode()
104 bool Decode(const VariantTensorData&) { in Decode()
Dlist_kernels.cc48 void TensorList::Encode(VariantTensorData* data) const { in Encode()
100 bool TensorList::Decode(const VariantTensorData& data) { in Decode()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dindexed_dataset_op.cc190 void Encode(VariantTensorData* data) const { in Encode()
195 bool Decode(const VariantTensorData& data) { in Decode()

12