Home
last modified time | relevance | path

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

12345

/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) { \
137 inline const int32* TensorProtoData<qint32>(const TensorProto& t) {
143 inline void Fill(const qint32* data, size_t n, TensorProto* t) {
159 inline const int* TensorProtoData<Eigen::half>(const TensorProto& t) {
[all …]
/external/tensorflow/tensorflow/core/framework/
Dtensor.cc275 static FieldType::const_iterator Begin(const TensorProto& proto) { \
278 static size_t NumElements(const TensorProto& proto) { \
281 static void Fill(const T* data, size_t n, TensorProto* proto) { \
304 static const int64* Begin(const TensorProto& proto) { in Begin()
307 static size_t NumElements(const TensorProto& proto) { in NumElements()
310 static void Fill(const int64* data, size_t n, TensorProto* proto) { in Fill()
318 static const uint64* Begin(const TensorProto& proto) { in Begin()
321 static size_t NumElements(const TensorProto& proto) { in NumElements()
324 static void Fill(const uint64* data, size_t n, TensorProto* proto) { in Fill()
333 const TensorProto& proto) { in Begin()
[all …]
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()
Dattr_value_util.cc51 string SummarizeTensor(const TensorProto& tensor_proto) { in SummarizeTensor()
391 void SetAttrValue(const TensorProto& value, AttrValue* out) { in SetAttrValue()
395 void SetAttrValue(const gtl::ArraySlice<TensorProto> value, AttrValue* out) { in SetAttrValue()
428 TensorProto ap; in AreAttrValuesEqual()
431 TensorProto bp; in AreAttrValuesEqual()
476 TensorProto p; in AttrValueHash()
Dattr_value_util.h67 void SetAttrValue(const TensorProto& value, AttrValue* out);
85 void SetAttrValue(gtl::ArraySlice<TensorProto> value, AttrValue* out);
Ddevice_base.h51 class TensorProto; variable
218 virtual Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
Dattr_value.proto25 repeated TensorProto tensor = 8; // "list(tensor)"
37 TensorProto tensor = 8; // "tensor"
Dtensor.h43 class TensorProto; variable
209 bool FromProto(const TensorProto& other) TF_MUST_USE_RESULT;
210 bool FromProto(Allocator* a, const TensorProto& other) TF_MUST_USE_RESULT;
217 void AsProtoField(TensorProto* proto) const;
218 void AsProtoTensorContent(TensorProto* proto) const;
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding.cc64 TensorProto empty; in InitFrom()
94 TensorProto empty; in ParseFrom()
147 protobuf::io::CodedInputStream* input, TensorProto* tensor_meta) { in ParseTensorSubmessage()
164 case TensorProto::kDtypeFieldNumber: { in ParseTensorSubmessage()
172 case TensorProto::kTensorShapeFieldNumber: { in ParseTensorSubmessage()
179 case TensorProto::kVersionNumberFieldNumber: { in ParseTensorSubmessage()
186 case TensorProto::kTensorContentFieldNumber: { in ParseTensorSubmessage()
286 TensorProto empty; in ParseSlow()
Dmessage_wrappers.cc26 bool ParseTensorProtoToTensor(const TensorProto& tensor_proto, in ParseTensorProtoToTensor()
67 TensorProto* out_tensor) const { in FeedValue()
160 TensorProto* out_tensor) const { in FeedValue()
169 TensorProto* value_proto = feed->mutable_tensor(); in add_feed()
246 Status ProtoRunStepRequest::FeedValue(size_t i, TensorProto* out_tensor) const { in FeedValue()
543 Status InMemoryRunGraphResponse::RecvValue(size_t i, TensorProto* out_tensor) { in RecvValue()
604 TensorProto* out_tensor) { in RecvValue()
621 TensorProto* value_proto = recv->mutable_tensor(); in AddRecv()
675 TensorProto* out_tensor) { in RecvValue()
692 TensorProto* value_proto = recv->mutable_tensor(); in AddRecv()
Dtensor_coding.h30 class TensorProto; variable
92 TensorProto* tensor_meta);
Dmessage_wrappers.h67 virtual Status FeedValue(size_t i, TensorProto* out_tensor) const = 0;
121 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
171 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
210 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
448 virtual Status RecvValue(size_t i, TensorProto* out_tensor) = 0;
486 Status RecvValue(size_t i, TensorProto* out_tensor) override;
519 Status RecvValue(size_t i, TensorProto* out_tensor) override;
546 Status RecvValue(size_t i, TensorProto* out_tensor) override;
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_tensor_coding.cc91 e->WriteUint64(TensorProto::kDtypeFieldNumber, val.dtype()); in EncodeSkeleton()
105 e->WriteVarlengthBeginning(TensorProto::kTensorShapeFieldNumber, in EncodeSkeleton()
121 TensorProto skeleton; in EncodeSkeleton()
126 TensorProto skeleton2; in EncodeSkeleton()
162 VarLengthEncodingSize(TensorProto::kTensorContentFieldNumber, in EncodeTensorToByteBuffer()
195 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/contrib/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/protobuf/
Dnamed_tensor.proto16 // The client can populate a TensorProto using a tensorflow::Tensor`, or
22 TensorProto tensor = 2;
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.h28 class TensorProto; variable
42 TensorProto* proto, bool is_dead,
/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.cc29 const TensorProto* proto = nullptr; in ConstOp()
108 TensorProto proto_;
/external/tensorflow/tensorflow/contrib/nccl/kernels/
Dnccl_rewrite.cc102 TensorProto TensorFromShape(const TensorShapeProto& shape) { in TensorFromShape()
103 TensorProto result; in TensorFromShape()
201 TensorProto tensor_proto = TensorFromShape(shape_proto); in ReplaceBroadcast()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc62 static std::vector<TensorProto> ExtractTensors(const AttrValue& attr_value) { in ExtractTensors()
63 std::vector<TensorProto> tensors; in ExtractTensors()
119 std::vector<TensorProto> tensors = ExtractTensors(attr_value); in ExtractExtraProperties()
122 const TensorProto& t = tensors[0]; in ExtractExtraProperties()

12345