Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/go/
Dtensor_test.go158 t2, err := ReadTensor(t1.DataType(), t1.Shape(), buf)
189 t3, err := ReadTensor(t1.DataType(), t1.Shape(), buf)
193 t4, err := ReadTensor(t2.DataType(), t2.Shape(), buf)
225 if _, err = ReadTensor(t1.DataType(), t1.Shape(), r); err == nil {
256 abts, err := ReadTensor(Float, []int64{2, 3}, abbuf)
Dtensor.go119 func ReadTensor(dataType DataType, shape []int64, r io.Reader) (*Tensor, error) { func
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils_test.cc62 TF_ASSERT_OK(reader.ReadTensor("Tensor", &val_tensor)); in TEST()
80 reader.ReadTensor("NonExistentKey", &val_tensor).code()); in TEST()
Dsparse_tensor_slice_dataset_op.cc189 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
191 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Ddataset_utils.h68 Status ReadTensor(StringPiece key, Tensor* val) override;
Dflat_map_dataset_op.cc235 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Ddataset_utils.cc170 Status VariantTensorDataReader::ReadTensor(StringPiece key, Tensor* val) { in ReadTensor() function in tensorflow::data::VariantTensorDataReader
Dinterleave_dataset_op.cc291 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreCurrentElements()
Dprefetch_dataset_op.cc212 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dparallel_interleave_dataset_op.cc807 TF_RETURN_IF_ERROR(reader->ReadTensor( in ReadElement()
828 TF_RETURN_IF_ERROR(reader->ReadTensor( in ReadElement()
Dparallel_map_iterator.cc171 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dwindow_dataset_op.cc303 reader->ReadTensor(strings::StrCat("buffer[", i, "][", j, "]"), in RestoreInternal()
Dshuffle_dataset_op.cc294 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dcache_dataset_ops.cc789 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunique_dataset_op.cc154 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dunbatch_dataset_op.cc201 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dparallel_interleave_dataset_op.cc856 TF_RETURN_IF_ERROR(reader->ReadTensor( in ReadWorkerStateLocked()
923 TF_RETURN_IF_ERROR(reader->ReadTensor( in ReadWorkerThreadStateLocked()
982 reader->ReadTensor(strings::StrCat(prefix, "_output_", i), in ReadOutputElemLocked()
Dsliding_window_dataset_op.cc280 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dscan_dataset_op.cc271 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dgroup_by_reducer_dataset_op.cc350 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
Dgroup_by_window_dataset_op.cc433 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreGroup()
Dmap_and_batch_dataset_op.cc684 TF_RETURN_IF_ERROR(reader->ReadTensor( in ReadBatchResult()
Dnuma_map_and_batch_dataset_op.cc711 TF_RETURN_IF_ERROR(reader->ReadTensor(tensor_name, &t)); in Restore()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h68 virtual Status ReadTensor(StringPiece key, Tensor* val) = 0;