Home
last modified time | relevance | path

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

1234567

/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) { \
140 inline const int32* TensorProtoData<qint32>(const TensorProto& t) {
146 inline void Fill(const qint32* data, size_t n, TensorProto* t) {
162 inline const int* TensorProtoData<Eigen::half>(const TensorProto& t) {
[all …]
/external/tensorflow/tensorflow/core/framework/
Dtensor_util.h80 std::declval<TensorProto>().FIELDNAME##_val(0)) FieldType; \
82 std::declval<TensorProto>().FIELDNAME##_val()) RepeatedFieldType; \
83 typedef decltype(std::declval<TensorProto>().mutable_##FIELDNAME##_val()) \
85 static MutableRepeatedFieldType GetMutableField(TensorProto* proto) { \
88 static RepeatedFieldType& GetField(const TensorProto& proto) { \
205 static size_t NumValues(const TensorProto& proto) {
210 static void AddValue(const T& value, TensorProto* proto) {
215 static T GetValue(size_t index, const TensorProto& proto) {
225 static void AddValues(IterType begin, IterType end, TensorProto* proto) {
232 static void CopyValues(IterType dst, const TensorProto& proto) {
[all …]
Dattr_value_util.cc43 int64 TensorByteSize(const TensorProto& t) { in TensorByteSize()
53 uint64 TensorProtoHash(const TensorProto& tp) { in TensorProtoHash()
57 TensorProto p; in TensorProtoHash()
66 uint64 FastTensorProtoHash(const TensorProto& tp) { in FastTensorProtoHash()
79 bool AreTensorProtosEqual(const TensorProto& lhs, const TensorProto& rhs) { in AreTensorProtosEqual()
88 TensorProto lhs_tp; in AreTensorProtosEqual()
91 TensorProto rhs_tp; in AreTensorProtosEqual()
100 bool FastAreTensorProtosEqual(const TensorProto& lhs, const TensorProto& rhs) { in FastAreTensorProtosEqual()
131 using TensorProtoHasher = std::function<uint64(const TensorProto&)>;
208 string SummarizeTensor(const TensorProto& tensor_proto) { in SummarizeTensor()
[all …]
Dtensor_util_test.cc438 TensorProto tensor_proto = in TEST()
460 TensorProto tensor_proto = in TEST()
521 TensorProto CreateAsProtoTensorContent(int size, int tail_length) { in CreateAsProtoTensorContent()
526 TensorProto tensor_proto; in CreateAsProtoTensorContent()
532 TensorProto CreateAsProtoField(int size, int tail_length) { in CreateAsProtoField()
537 TensorProto tensor_proto; in CreateAsProtoField()
543 void CompareTensorValues(const TensorProto& x, const TensorProto& y) { in CompareTensorValues()
557 TensorProto tensor_proto = in ConstantTailTest()
560 TensorProto original_tensor_proto = tensor_proto; in ConstantTailTest()
/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()
293 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;
126 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
177 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
218 Status FeedValue(size_t i, TensorProto* out_tensor) const override;
485 virtual Status RecvValue(size_t i, TensorProto* out_tensor) = 0;
523 Status RecvValue(size_t i, TensorProto* out_tensor) override;
556 Status RecvValue(size_t i, TensorProto* out_tensor) override;
583 Status RecvValue(size_t i, TensorProto* out_tensor) override;
740 bool ParseTensorProtoToTensor(const TensorProto& tensor_proto,
Dmessage_wrappers.cc26 bool ParseTensorProtoToTensor(const TensorProto& tensor_proto, in ParseTensorProtoToTensor()
65 TensorProto* out_tensor) const { in FeedValue()
162 TensorProto* out_tensor) const { in FeedValue()
171 TensorProto* value_proto = feed->mutable_tensor(); in add_feed()
252 Status ProtoRunStepRequest::FeedValue(size_t i, TensorProto* out_tensor) const { in FeedValue()
621 Status InMemoryRunGraphResponse::RecvValue(size_t i, TensorProto* out_tensor) { in RecvValue()
682 TensorProto* out_tensor) { in RecvValue()
699 TensorProto* value_proto = recv->mutable_tensor(); in AddRecv()
753 TensorProto* out_tensor) { in RecvValue()
770 TensorProto* value_proto = recv->mutable_tensor(); in AddRecv()
/external/tensorflow/tensorflow/core/kernels/
Dunique_op_test.cc40 TensorProto GetRandomInt32TensorProto(int dim, int max_int) { in GetRandomInt32TensorProto()
41 TensorProto tensor_proto; in GetRandomInt32TensorProto()
52 TensorProto GetRandomInt32TensorProtoWithRepeat(int dim, int repeat, in GetRandomInt32TensorProtoWithRepeat()
54 TensorProto tensor_proto; in GetRandomInt32TensorProtoWithRepeat()
114 TensorProto GetRandomStringsTensorProto(int dim, int max_str_len) { in GetRandomStringsTensorProto()
115 TensorProto tensor_proto; in GetRandomStringsTensorProto()
Dparse_tensor_op.cc44 TensorProto proto; in Compute()
76 TensorProto proto; in Compute()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_tensor_coding.cc95 e->WriteUint64(TensorProto::kDtypeFieldNumber, val.dtype()); in EncodeSkeleton()
109 e->WriteVarlengthBeginning(TensorProto::kTensorShapeFieldNumber, in EncodeSkeleton()
125 TensorProto skeleton; in EncodeSkeleton()
130 TensorProto skeleton2; in EncodeSkeleton()
176 VarLengthEncodingSize(TensorProto::kTensorContentFieldNumber, in EncodeTensorToByteBuffer()
214 e.WriteVarlengthBeginning(TensorProto::kTensorContentFieldNumber, in EncodeTensorToByteBuffer()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_utils.cc73 DataType dtype, const std::function<void(TensorProto*)>& add_value, in AddScalarConstNodeHelper()
80 std::unique_ptr<tensorflow::TensorProto> tensor = in AddScalarConstNodeHelper()
81 tensorflow::MakeUnique<tensorflow::TensorProto>(); in AddScalarConstNodeHelper()
127 DT_BOOL, [v](TensorProto* proto) { proto->add_bool_val(v); }, graph); in AddScalarConstNode()
133 DT_DOUBLE, [v](TensorProto* proto) { proto->add_double_val(v); }, graph); in AddScalarConstNode()
139 DT_FLOAT, [v](TensorProto* proto) { proto->add_float_val(v); }, graph); in AddScalarConstNode()
145 DT_INT32, [v](TensorProto* proto) { proto->add_int_val(v); }, graph); in AddScalarConstNode()
151 DT_INT64, [v](TensorProto* proto) { proto->add_int64_val(v); }, graph); in AddScalarConstNode()
158 [v](TensorProto* proto) { proto->add_string_val(v.data(), v.size()); }, in AddScalarConstNode()
/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/python/framework/
Dop_def_util_test.py54 ("tensor", tensor_pb2.TensorProto(dtype=types_pb2.DataType.DT_FLOAT),
55 tensor_pb2.TensorProto(dtype=types_pb2.DataType.DT_FLOAT)),
57 tensor_pb2.TensorProto(dtype=types_pb2.DataType.DT_FLOAT)),
/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/compiler/mlir/tensorflow/utils/
Dmangling_util.h49 string MangleTensor(const TensorProto& tensor);
51 Status DemangleTensor(absl::string_view str, TensorProto* proto);
Dconvert_tensor.cc61 static TensorProto ConvertToProto(const Tensor& input_tensor, in ConvertToProto()
63 TensorProto tensor_proto; in ConvertToProto()
171 StatusOr<ElementsAttr> ConvertTensorProto(const TensorProto& input_tensor, in ConvertTensorProto()
251 TensorProto* output_tensor) { in ConvertOpaqueElementsAttr()
309 Status ConvertToTensorProto(const ElementsAttr attr, TensorProto* output) { in ConvertToTensorProto()
376 TensorProto tensor_proto; in ConvertToTensor()
Dconvert_tensor.h34 StatusOr<mlir::ElementsAttr> ConvertTensorProto(const TensorProto& input_tensor,
57 TensorProto* output_tensor);
Dmangling_util.cc71 string MangleTensor(const TensorProto& tensor) { in MangleTensor()
75 Status DemangleTensor(absl::string_view str, TensorProto* proto) { in DemangleTensor()
/external/tensorflow/tensorflow/core/protobuf/
Dnamed_tensor.proto18 // The client can populate a TensorProto using a tensorflow::Tensor`, or
24 TensorProto tensor = 2;
/external/tensorflow/tensorflow/core/data/
Dcompression_utils.cc28 std::vector<TensorProto> non_memcpy_components; in CompressElement()
59 TensorProto& proto = non_memcpy_components[non_memcpy_component_index++]; in CompressElement()
138 TensorProto tp; in UncompressElement()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.h28 class TensorProto; variable
40 TensorProto* proto, bool is_dead,
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.h155 Status MakeTensorFromProto(const TensorProto& tensor_proto,
162 Status MakeFastMemTensorFromProto(const TensorProto& tensor_proto,
206 const TensorProto& tensor_proto,

1234567