/external/tensorflow/tensorflow/core/util/ |
D | saved_tensor_slice_util.h | 64 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/ |
D | tensor_util.h | 76 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 …]
|
D | tensor.cc | 283 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 …]
|
D | attr_value_util.cc | 41 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 …]
|
D | tensor_util_test.cc | 437 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()
|
D | variant_test.cc | 104 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()
|
D | tensor_util.cc | 187 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/ |
D | conv_autotuning.proto | 12 TensorProto input = 2; 13 TensorProto filter = 3; 14 TensorProto output = 4; 15 TensorProto bias = 5; 17 TensorProto side_input = 6;
|
D | named_tensor.proto | 16 // The client can populate a TensorProto using a tensorflow::Tensor`, or 22 TensorProto tensor = 2;
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | tensor_coding.cc | 64 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()
|
D | tensor_coding.h | 30 class TensorProto; variable 96 TensorProto* tensor_meta);
|
D | message_wrappers.h | 67 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;
|
D | message_wrappers.cc | 26 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/ |
D | grpc_tensor_coding.cc | 93 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/ |
D | unique_op_test.cc | 38 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()
|
D | parse_tensor_op.cc | 44 TensorProto proto; in Compute() 76 TensorProto proto; in Compute()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | graph_utils.cc | 71 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/ |
D | resolve_svdf_test.cc | 105 tensorflow::TensorProto* allocated_tensor = new tensorflow::TensorProto; in AddNewNode() 124 tensorflow::TensorProto* allocated_tensor = new tensorflow::TensorProto; in AddShapeNode()
|
D | resolve_svdf.cc | 64 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/ |
D | api_def_SerializeTensor.pbtxt | 12 A serialized TensorProto proto of the input tensor. 21 summary: "Transforms a Tensor into a serialized TensorProto proto."
|
D | api_def_ParseTensor.pbtxt | 6 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/ |
D | gpu_util.h | 28 class TensorProto; variable 40 TensorProto* proto, bool is_dead,
|
/external/tensorflow/tensorflow/lite/toco/ |
D | import_tensorflow.cc | 64 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/ |
D | rdma.h | 342 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/ |
D | const_op.cc | 31 const TensorProto* proto = nullptr; in ConstOp() 127 TensorProto proto_;
|