Home
last modified time | relevance | path

Searched refs:TensorProto (Results 1 – 25 of 134) sorted by relevance

123456

/external/tensorflow/tensorflow/core/util/
Dsaved_tensor_slice_util.h64 const TensorProto& t);
68 TensorProto* t);
71 void Fill(T* data, size_t n, TensorProto* t);
81 inline const STYPE* TensorProtoData<TYPE>(const TensorProto& t) { \
88 TensorProto * t) { \
98 inline void Fill(const TYPE* data, size_t n, TensorProto* t) { \
107 inline void Fill(const TYPE* data, size_t n, TensorProto* t) { \
138 inline const int32* TensorProtoData<qint32>(const TensorProto& t) {
144 inline void Fill(const qint32* data, size_t n, TensorProto* t) {
160 inline const int* TensorProtoData<Eigen::half>(const TensorProto& t) {
[all …]
/external/tensorflow/tensorflow/core/framework/
Dtensor_util.h76 std::declval<TensorProto>().FIELDNAME##_val(0)) FieldType; \
78 std::declval<TensorProto>().FIELDNAME##_val()) RepeatedFieldType; \
79 typedef decltype(std::declval<TensorProto>().mutable_##FIELDNAME##_val()) \
81 static MutableRepeatedFieldType GetMutableField(TensorProto* proto) { \
84 static RepeatedFieldType& GetField(const TensorProto& proto) { \
203 static size_t NumValues(const TensorProto& proto) {
208 static void AddValue(const T& value, TensorProto* proto) {
213 static T GetValue(size_t index, const TensorProto& proto) {
223 static void AddValues(IterType begin, IterType end, TensorProto* proto) {
230 static void CopyValues(IterType dst, const TensorProto& proto) {
[all …]
Dtensor.cc283 static FieldType::const_iterator Begin(const TensorProto& proto) { \
286 static size_t NumElements(const TensorProto& proto) { \
289 static void Fill(const T* data, size_t n, TensorProto* proto) { \
312 static const int64* Begin(const TensorProto& proto) { in Begin()
315 static size_t NumElements(const TensorProto& proto) { in NumElements()
318 static void Fill(const int64* data, size_t n, TensorProto* proto) { in Fill()
326 static const uint64* Begin(const TensorProto& proto) { in Begin()
329 static size_t NumElements(const TensorProto& proto) { in NumElements()
332 static void Fill(const uint64* data, size_t n, TensorProto* proto) { in Fill()
341 const TensorProto& proto) { in Begin()
[all …]
Dattr_value_util.cc41 int64 TensorByteSize(const TensorProto& t) { in TensorByteSize()
51 uint64 TensorProtoHash(const TensorProto& tp) { in TensorProtoHash()
55 TensorProto p; in TensorProtoHash()
64 uint64 FastTensorProtoHash(const TensorProto& tp) { in FastTensorProtoHash()
77 bool AreTensorProtosEqual(const TensorProto& lhs, const TensorProto& rhs) { in AreTensorProtosEqual()
86 TensorProto lhs_tp; in AreTensorProtosEqual()
89 TensorProto rhs_tp; in AreTensorProtosEqual()
98 bool FastAreTensorProtosEqual(const TensorProto& lhs, const TensorProto& rhs) { in FastAreTensorProtosEqual()
113 using TensorProtoHasher = std::function<uint64(const TensorProto&)>;
115 std::function<bool(const TensorProto&, const TensorProto&)>;
[all …]
Dtensor_util_test.cc437 TensorProto tensor_proto = in TEST()
459 TensorProto tensor_proto = in TEST()
508 TensorProto CreateAsProtoTensorContent(int size, int tail_length) { in CreateAsProtoTensorContent()
512 TensorProto tensor_proto; in CreateAsProtoTensorContent()
518 TensorProto CreateAsProtoField(int size, int tail_length) { in CreateAsProtoField()
522 TensorProto tensor_proto; in CreateAsProtoField()
528 void CompareTensorValues(const TensorProto& x, const TensorProto& y) { in CompareTensorValues()
542 TensorProto tensor_proto = in ConstantTailTest()
545 TensorProto original_tensor_proto = tensor_proto; in ConstantTailTest()
Dvariant_test.cc104 TEST(VariantTest, TensorProto) { in TEST() argument
106 TensorProto t; in TEST()
112 EXPECT_NE(x.get<TensorProto>(), nullptr); in TEST()
113 EXPECT_EQ(x.get<TensorProto>()->dtype(), DT_FLOAT); in TEST()
114 EXPECT_EQ(x.get<TensorProto>()->tensor_shape().unknown_rank(), true); in TEST()
Dtensor_util.cc187 const TensorShape& shape, TensorProto* tensor) { in CompressTensorContent()
266 const TensorShape& shape, TensorProto* tensor) { in CompressRepeatedField()
311 TensorProto* tensor) { in CompressTensorProtoInPlaceImpl()
336 TensorProto* tensor) { in CompressTensorProtoInPlace()
/external/tensorflow/tensorflow/core/protobuf/
Dconv_autotuning.proto12 TensorProto input = 2;
13 TensorProto filter = 3;
14 TensorProto output = 4;
15 TensorProto bias = 5;
17 TensorProto side_input = 6;
Dnamed_tensor.proto16 // The client can populate a TensorProto using a tensorflow::Tensor`, or
22 TensorProto tensor = 2;
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding.cc64 TensorProto empty; in InitFrom()
95 TensorProto empty; in ParseFrom()
148 protobuf::io::CodedInputStream* input, TensorProto* tensor_meta) { in ParseTensorSubmessage()
165 case TensorProto::kDtypeFieldNumber: { in ParseTensorSubmessage()
173 case TensorProto::kTensorShapeFieldNumber: { in ParseTensorSubmessage()
180 case TensorProto::kVersionNumberFieldNumber: { in ParseTensorSubmessage()
187 case TensorProto::kTensorContentFieldNumber: { in ParseTensorSubmessage()
287 TensorProto empty; in ParseSlow()
Dtensor_coding.h30 class TensorProto; variable
96 TensorProto* tensor_meta);
Dmessage_wrappers.h67 virtual Status FeedValue(size_t i, TensorProto* out_tensor) const = 0;
123 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
174 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
215 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
473 virtual Status RecvValue(size_t i, TensorProto* out_tensor) = 0;
511 Status RecvValue(size_t i, TensorProto* out_tensor) override;
544 Status RecvValue(size_t i, TensorProto* out_tensor) override;
571 Status RecvValue(size_t i, TensorProto* out_tensor) override;
Dmessage_wrappers.cc26 bool ParseTensorProtoToTensor(const TensorProto& tensor_proto, in ParseTensorProtoToTensor()
67 TensorProto* out_tensor) const { in FeedValue()
164 TensorProto* out_tensor) const { in FeedValue()
173 TensorProto* value_proto = feed->mutable_tensor(); in add_feed()
254 Status ProtoRunStepRequest::FeedValue(size_t i, TensorProto* out_tensor) const { in FeedValue()
602 Status InMemoryRunGraphResponse::RecvValue(size_t i, TensorProto* out_tensor) { in RecvValue()
663 TensorProto* out_tensor) { in RecvValue()
680 TensorProto* value_proto = recv->mutable_tensor(); in AddRecv()
734 TensorProto* out_tensor) { in RecvValue()
751 TensorProto* value_proto = recv->mutable_tensor(); in AddRecv()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_tensor_coding.cc93 e->WriteUint64(TensorProto::kDtypeFieldNumber, val.dtype()); in EncodeSkeleton()
107 e->WriteVarlengthBeginning(TensorProto::kTensorShapeFieldNumber, in EncodeSkeleton()
123 TensorProto skeleton; in EncodeSkeleton()
128 TensorProto skeleton2; in EncodeSkeleton()
164 VarLengthEncodingSize(TensorProto::kTensorContentFieldNumber, in EncodeTensorToByteBuffer()
202 e.WriteVarlengthBeginning(TensorProto::kTensorContentFieldNumber, in EncodeTensorToByteBuffer()
/external/tensorflow/tensorflow/core/kernels/
Dunique_op_test.cc38 TensorProto GetRandomInt32TensorProto(int dim, int max_int) { in GetRandomInt32TensorProto()
39 TensorProto tensor_proto; in GetRandomInt32TensorProto()
50 TensorProto GetRandomInt32TensorProtoWithRepeat(int dim, int repeat, in GetRandomInt32TensorProtoWithRepeat()
52 TensorProto tensor_proto; in GetRandomInt32TensorProtoWithRepeat()
105 TensorProto GetRandomStringsTensorProto(int dim, int max_str_len) { in GetRandomStringsTensorProto()
106 TensorProto tensor_proto; in GetRandomStringsTensorProto()
Dparse_tensor_op.cc44 TensorProto proto; in Compute()
76 TensorProto proto; in Compute()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_utils.cc71 DataType dtype, const std::function<void(TensorProto*)>& add_value, in AddScalarConstNodeHelper()
78 std::unique_ptr<tensorflow::TensorProto> tensor = in AddScalarConstNodeHelper()
79 tensorflow::MakeUnique<tensorflow::TensorProto>(); in AddScalarConstNodeHelper()
125 DT_BOOL, [v](TensorProto* proto) { proto->add_bool_val(v); }, graph); in AddScalarConstNode()
131 DT_DOUBLE, [v](TensorProto* proto) { proto->add_double_val(v); }, graph); in AddScalarConstNode()
137 DT_FLOAT, [v](TensorProto* proto) { proto->add_float_val(v); }, graph); in AddScalarConstNode()
143 DT_INT32, [v](TensorProto* proto) { proto->add_int_val(v); }, graph); in AddScalarConstNode()
149 DT_INT64, [v](TensorProto* proto) { proto->add_int64_val(v); }, graph); in AddScalarConstNode()
156 [v](TensorProto* proto) { proto->add_string_val(v.data(), v.size()); }, in AddScalarConstNode()
/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_svdf_test.cc105 tensorflow::TensorProto* allocated_tensor = new tensorflow::TensorProto; in AddNewNode()
124 tensorflow::TensorProto* allocated_tensor = new tensorflow::TensorProto; in AddShapeNode()
Dresolve_svdf.cc64 const tensorflow::TensorProto& tensor = value_attr.tensor(); in FilterPartitionedConstNodes()
84 const tensorflow::TensorProto& tensor = value_attr.tensor(); in InferFilterRank()
183 tensorflow::TensorProto* allocated_tensor = in MaybeMergeConstNodes()
191 const tensorflow::TensorProto& tensor = value_attr.tensor(); in MaybeMergeConstNodes()
219 const tensorflow::TensorProto& tensor = value_attr.tensor(); in MaybeMergeConstNodes()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SerializeTensor.pbtxt12 A serialized TensorProto proto of the input tensor.
21 summary: "Transforms a Tensor into a serialized TensorProto proto."
Dapi_def_ParseTensor.pbtxt6 A scalar string containing a serialized TensorProto proto.
22 summary: "Transforms a serialized tensorflow.TensorProto proto into a Tensor."
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.h28 class TensorProto; variable
40 TensorProto* proto, bool is_dead,
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc64 using tensorflow::TensorProto;
126 const TensorProto& GetTensorAttr(const NodeDef& node, const string& attr_name) { in GetTensorAttr()
245 static int size(const TensorProto& p) { return p.float_val_size(); } in size()
246 static float get(const TensorProto& p, int i) { return p.float_val(i); } in get()
249 static void CopyFromContent(const TensorProto& p, std::vector<float>* data) { in CopyFromContent()
257 static int size(const TensorProto& p) { return p.int_val_size(); } in size()
258 static uint8_t get(const TensorProto& p, int i) { return p.int_val(i); } in get()
261 static void CopyFromContent(const TensorProto& p, in CopyFromContent()
270 static int size(const TensorProto& p) { return p.scomplex_val_size() / 2; } in size()
271 static std::complex<float> get(const TensorProto& p, int i) { in get()
[all …]
/external/tensorflow/tensorflow/contrib/verbs/
Drdma.h342 void Clone(const Tensor& in, const TensorProto& proto, bool is_dead);
343 void Send(const Tensor& in, const TensorProto& proto, bool is_dead,
348 void SendMetaData(const Tensor& in, const TensorProto& proto, bool is_dead);
349 void SendContent(const Tensor& in, const TensorProto& proto, bool is_dead);
362 TensorProto* proto_ = nullptr;
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconst_op.cc31 const TensorProto* proto = nullptr; in ConstOp()
127 TensorProto proto_;

123456