Home
last modified time | relevance | path

Searched refs:WriteTensor (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/summary/
Dsummary_db_writer_test.cc181 TF_ASSERT_OK(writer_->WriteTensor(1, MakeScalarInt64(123LL), "taggy", in TEST_F()
184 TF_ASSERT_OK(writer_->WriteTensor(2, MakeScalarInt64(314LL), "taggy", in TEST_F()
235 TF_ASSERT_OK(writer_->WriteTensor(1, MakeScalarInt64(123LL), "taggy", "")); in TEST_F()
359 TF_ASSERT_OK(writer_->WriteTensor(1, MakeScalarInt64(123LL), "taggy", in TEST_F()
376 TF_ASSERT_OK(writer_->WriteTensor(1, MakeScalarInt64(123LL), "taggy", in TEST_F()
Dsummary_file_writer_test.cc92 TEST_F(SummaryFileWriterTest, WriteTensor) { in TEST_F() argument
97 TF_RETURN_IF_ERROR(writer->WriteTensor( in TEST_F()
113 TF_RETURN_IF_ERROR(writer->WriteTensor( in TEST_F()
Dsummary_file_writer.cc70 Status WriteTensor(int64 global_step, Tensor t, const string& tag, in WriteTensor() function in tensorflow::__anon5ecb8bc70111::SummaryFileWriter
/external/tensorflow/tensorflow/core/kernels/
Dsummary_interface.h38 virtual Status WriteTensor(int64 global_step, Tensor t, const string& tag,
Dsummary_kernels.cc142 OP_REQUIRES_OK(ctx, s->WriteTensor(step, *t, tag, serialized_metadata)); in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils_test.cc52 TF_ASSERT_OK(writer.WriteTensor(full_name("Tensor"), input_tensor)); in TEST()
89 TF_ASSERT_OK(writer.WriteTensor("Iterator", "Tensor", input_tensor)); in TEST()
131 writer.WriteTensor(full_name("Tensor"), input_tensor).code()); in TEST()
Ddataset_utils.cc67 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteElementsToCheckpoint()
323 Status VariantTensorDataWriter::WriteTensor(StringPiece key, in WriteTensor() function in tensorflow::data::VariantTensorDataWriter
338 Status VariantTensorDataWriter::WriteTensor(StringPiece name, StringPiece key, in WriteTensor() function in tensorflow::data::VariantTensorDataWriter
757 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteBatch()
762 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteBatch()
Dsparse_tensor_slice_dataset_op.cc177 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
179 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Ddataset_utils.h209 Status WriteTensor(StringPiece key, const Tensor& val) override;
215 Status WriteTensor(StringPiece name, StringPiece key,
Dflat_map_dataset_op.cc185 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dinterleave_dataset_op.cc254 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveCurrentElements()
Dwindow_dataset_op.cc283 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dprefetch_dataset_op.cc253 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dparallel_interleave_dataset_op.cc1190 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteElement()
1207 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteElement()
Dparallel_map_dataset_op.cc301 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Drebatch_dataset_op.cc198 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
535 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dunique_dataset_op.cc124 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dunbatch_dataset_op.cc203 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dparallel_interleave_dataset_op.cc910 TF_RETURN_IF_ERROR(writer->WriteTensor(iterator_name, in WriteWorkerStateLocked()
978 writer->WriteTensor(iterator_name, strings::StrCat(kInput, "_", i), in WriteWorkerThreadStateLocked()
1047 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteOutputElemLocked()
Dsliding_window_dataset_op.cc264 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dscan_dataset_op.cc273 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dgroup_by_reducer_dataset_op.cc321 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dgroup_by_window_dataset_op.cc446 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveGroup()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h119 virtual Status WriteTensor(StringPiece key, const Tensor& val) = 0;
125 virtual Status WriteTensor(StringPiece name, StringPiece key,
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc228 Status WriteTensor(const Tensor& val, FileOutputBuffer* out, in WriteTensor() function
463 status_ = WriteTensor(val, out_.get(), &data_bytes_written); in Add()

12