Home
last modified time | relevance | path

Searched refs:WriteTensor (Results 1 – 25 of 29) 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::__anonf5c561500111::SummaryFileWriter
/external/tensorflow/tensorflow/core/kernels/
Dsummary_interface.h38 virtual Status WriteTensor(int64 global_step, Tensor t, const string& tag,
Dsummary_kernels.cc144 OP_REQUIRES_OK(ctx, s->WriteTensor(step, *t, tag, serialized_metadata)); in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Dsparse_tensor_slice_dataset_op.cc170 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
172 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Ddataset_utils.h87 Status WriteTensor(StringPiece key, const Tensor& val) override;
Ddataset_utils_test.cc54 TF_ASSERT_OK(writer.WriteTensor("Tensor", input_tensor)); in TEST()
Dflat_map_dataset_op.cc191 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Ddataset_utils.cc205 Status VariantTensorDataWriter::WriteTensor(StringPiece key, in WriteTensor() function in tensorflow::data::VariantTensorDataWriter
Dinterleave_dataset_op.cc270 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveCurrentElements()
Dprefetch_dataset_op.cc176 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dparallel_interleave_dataset_op.cc718 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteElement()
738 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteElement()
Dparallel_map_iterator.cc130 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dwindow_dataset_op.cc275 writer->WriteTensor(strings::StrCat("buffer[", i, "][", j, "]"), in SaveInternal()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunique_dataset_op.cc134 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dunbatch_dataset_op.cc178 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dparallel_interleave_dataset_op.cc825 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteWorkerStateLocked()
894 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteWorkerThreadStateLocked()
964 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteOutputElemLocked()
Dsliding_window_dataset_op.cc254 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dscan_dataset_op.cc254 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dmap_and_batch_dataset_op.cc752 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteBatchResult()
756 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteBatchResult()
Dnuma_map_and_batch_dataset_op.cc653 TF_RETURN_IF_ERROR(writer->WriteTensor( in Save()
657 TF_RETURN_IF_ERROR(writer->WriteTensor( in Save()
Dgroup_by_reducer_dataset_op.cc299 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
Dgroup_by_window_dataset_op.cc413 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveGroup()
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc212 Status WriteTensor(const Tensor& val, FileOutputBuffer* out, in WriteTensor() function
436 status_ = WriteTensor(val, out_.get(), &data_bytes_written); in Add()

12