/external/tensorflow/tensorflow/go/ |
D | tensor_test.go | 158 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)
|
D | tensor.go | 119 func ReadTensor(dataType DataType, shape []int64, r io.Reader) (*Tensor, error) { func
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | dataset_utils_test.cc | 62 TF_ASSERT_OK(reader.ReadTensor("Tensor", &val_tensor)); in TEST() 80 reader.ReadTensor("NonExistentKey", &val_tensor).code()); in TEST()
|
D | sparse_tensor_slice_dataset_op.cc | 189 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal() 191 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
D | dataset_utils.h | 68 Status ReadTensor(StringPiece key, Tensor* val) override;
|
D | flat_map_dataset_op.cc | 235 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
D | dataset_utils.cc | 170 Status VariantTensorDataReader::ReadTensor(StringPiece key, Tensor* val) { in ReadTensor() function in tensorflow::data::VariantTensorDataReader
|
D | interleave_dataset_op.cc | 291 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreCurrentElements()
|
D | prefetch_dataset_op.cc | 212 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
D | parallel_interleave_dataset_op.cc | 807 TF_RETURN_IF_ERROR(reader->ReadTensor( in ReadElement() 828 TF_RETURN_IF_ERROR(reader->ReadTensor( in ReadElement()
|
D | parallel_map_iterator.cc | 171 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
D | window_dataset_op.cc | 303 reader->ReadTensor(strings::StrCat("buffer[", i, "][", j, "]"), in RestoreInternal()
|
D | shuffle_dataset_op.cc | 294 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
D | cache_dataset_ops.cc | 789 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | unique_dataset_op.cc | 154 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
D | unbatch_dataset_op.cc | 201 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
D | parallel_interleave_dataset_op.cc | 856 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()
|
D | sliding_window_dataset_op.cc | 280 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
D | scan_dataset_op.cc | 271 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
D | group_by_reducer_dataset_op.cc | 350 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreInternal()
|
D | group_by_window_dataset_op.cc | 433 TF_RETURN_IF_ERROR(reader->ReadTensor( in RestoreGroup()
|
D | map_and_batch_dataset_op.cc | 684 TF_RETURN_IF_ERROR(reader->ReadTensor( in ReadBatchResult()
|
D | numa_map_and_batch_dataset_op.cc | 711 TF_RETURN_IF_ERROR(reader->ReadTensor(tensor_name, &t)); in Restore()
|
/external/tensorflow/tensorflow/core/framework/ |
D | dataset.h | 68 virtual Status ReadTensor(StringPiece key, Tensor* val) = 0;
|