Home
last modified time | relevance | path

Searched refs:ReadTensor (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/core/data/
Dserialization_utils.h56 Status ReadTensor(StringPiece key, Tensor* val) const override;
57 Status ReadTensor(FunctionLibraryRuntime* flr, StringPiece key,
59 Status ReadTensor(StringPiece name, StringPiece key,
61 Status ReadTensor(FunctionLibraryRuntime* flr, StringPiece name,
Dserialization_utils_test.cc142 TF_ASSERT_OK(reader.ReadTensor(full_name("Tensor"), &val_tensor)); in TEST()
162 reader.ReadTensor(full_name("NonExistentKey"), &val_tensor).code()); in TEST()
179 TF_ASSERT_OK(reader.ReadTensor("Iterator", "Tensor", &val_tensor)); in TEST()
201 reader.ReadTensor("Iterator", "NonExistentKey", &val_tensor).code()); in TEST()
Dserialization_utils.cc124 TF_RETURN_IF_ERROR(reader->ReadTensor( in ReadElementsFromCheckpoint()
188 Status VariantTensorDataReader::ReadTensor(StringPiece key, Tensor* val) const { in ReadTensor() function in tensorflow::data::VariantTensorDataReader
191 return ReadTensor(name, key, val); in ReadTensor()
194 Status VariantTensorDataReader::ReadTensor(FunctionLibraryRuntime* flr, in ReadTensor() function in tensorflow::data::VariantTensorDataReader
201 Status VariantTensorDataReader::ReadTensor(StringPiece name, StringPiece key, in ReadTensor() function in tensorflow::data::VariantTensorDataReader
203 return ReadTensor(/*flr=*/nullptr, name, key, val); in ReadTensor()
206 Status VariantTensorDataReader::ReadTensor(FunctionLibraryRuntime* flr, in ReadTensor() function in tensorflow::data::VariantTensorDataReader
Ddataset_utils.cc599 reader->ReadTensor(ctx->flr(), FullName(iterator_prefix, batch_prefix), in ReadBatch()
/external/tensorflow/tensorflow/go/
Dtensor_test.go159 t2, err := ReadTensor(t1.DataType(), t1.Shape(), buf)
190 t3, err := ReadTensor(t1.DataType(), t1.Shape(), buf)
194 t4, err := ReadTensor(t2.DataType(), t2.Shape(), buf)
226 if _, err = ReadTensor(t1.DataType(), t1.Shape(), r); err == nil {
257 abts, err := ReadTensor(Float, []int64{2, 3}, abbuf)
281 _, err := ReadTensor(Int32, []int64{-1, 1}, buf)
Dtensor.go175 func ReadTensor(dataType DataType, shape []int64, r io.Reader) (*Tensor, error) { func
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc66 RETURN_IF_ERROR(reader->ReadTensor(weights_tensor_id, &weights)); in GetFullyConnectedAttributes()
73 reader->ReadTensor(bias_tensor_id, &attr->bias).IgnoreError(); // optional in GetFullyConnectedAttributes()
151 RETURN_IF_ERROR(reader->ReadTensor(constant_tensor, &tensor)); in ParseInputsWithConstTensor()
156 RETURN_IF_ERROR(reader->ReadTensor(constant_tensor, &tensor)); in ParseInputsWithConstTensor()
160 RETURN_IF_ERROR(reader->ReadTensor(constant_tensor, &tensor)); in ParseInputsWithConstTensor()
397 RETURN_IF_ERROR(reader->ReadTensor(idx, &tensor)); in Parse()
505 RETURN_IF_ERROR(reader->ReadTensor(1, &attr.weights)); in Parse()
507 reader->ReadTensor(2, &attr.bias).IgnoreError(); // bias is optional in Parse()
542 RETURN_IF_ERROR(reader->ReadTensor(0, &sparse_tensor)); in Parse()
572 RETURN_IF_ERROR(reader->ReadTensor(1, &attr.weights)); in Parse()
[all …]
Dlstm_parser.cc79 reader->ReadTensor(bias_tensor_id, &(fc_attr.bias)).IgnoreError(); in GetFullyConnectedNode()
86 RETURN_IF_ERROR(reader->ReadTensor(weights_tensor_id, &weights)); in GetFullyConnectedNode()
94 reader->ReadTensor(bias_tensor_id, &(fc_attr.bias)).IgnoreError(); in GetFullyConnectedNode()
187 RETURN_IF_ERROR(reader->ReadTensor(cell_weight_id, &weights)); in BuildLstmGate()
236 RETURN_IF_ERROR(reader->ReadTensor(normalization_weight_id, &norm_weights)); in BuildLstmGate()
250 RETURN_IF_ERROR(reader->ReadTensor(bias_id, &bias)); in BuildLstmGate()
Dobject_reader.h65 absl::Status ReadTensor(uint32_t index, TensorT* tensor) const { in ReadTensor() function
/external/tensorflow/tensorflow/core/kernels/data/
Dsparse_tensor_slice_dataset_op.cc196 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
198 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dflat_map_dataset_op.cc273 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dinterleave_dataset_op.cc318 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreCurrentElements()
Dwindow_dataset_op.cc311 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dprefetch_dataset_op.cc295 reader->ReadTensor(ctx->flr(), absl::StrCat(prefix(), "::", i), in RestoreInternal()
Dparallel_interleave_dataset_op.cc1300 TF_RETURN_IF_ERROR(reader->ReadTensor( in ReadElement()
1321 reader->ReadTensor(ctx->flr(), iterator_name, in ReadElement()
Dparallel_map_dataset_op.cc341 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h118 virtual Status ReadTensor(StringPiece key, Tensor* val) const = 0;
119 virtual Status ReadTensor(FunctionLibraryRuntime* flr, StringPiece key,
121 virtual Status ReadTensor(StringPiece name, StringPiece key,
123 virtual Status ReadTensor(FunctionLibraryRuntime* flr, StringPiece name,
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunique_dataset_op.cc144 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Drebatch_dataset_op.cc227 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
571 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dunbatch_dataset_op.cc232 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dsliding_window_dataset_op.cc290 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dparallel_interleave_dataset_op.cc949 TF_RETURN_IF_ERROR(reader->ReadTensor(ctx->flr(), worker_prefix, in ReadWorkerStateLocked()
1015 TF_RETURN_IF_ERROR(reader->ReadTensor(ctx->flr(), worker_prefix, in ReadWorkerThreadStateLocked()
1079 reader->ReadTensor(ctx->flr(), iterator_name, in ReadOutputElemLocked()
Dscan_dataset_op.cc290 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dgroup_by_reducer_dataset_op.cc372 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dgroup_by_window_dataset_op.cc467 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreGroup()

12