Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/summary/
H A 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()
H A Dsummary_file_writer_test.cc93 TEST_F(SummaryFileWriterTest, WriteTensor) { in TEST_F() argument
98 TF_RETURN_IF_ERROR(writer->WriteTensor( in TEST_F()
114 TF_RETURN_IF_ERROR(writer->WriteTensor( in TEST_F()
H A Dsummary_file_writer.cc80 Status WriteTensor(int64_t global_step, Tensor t, const string& tag, in WriteTensor() function in tensorflow::__anon942ebc7d0111::SummaryFileWriter
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dserialization_utils_test.cc133 TF_ASSERT_OK(writer.WriteTensor(full_name("Tensor"), input_tensor)); in TEST()
170 TF_ASSERT_OK(writer.WriteTensor("Iterator", "Tensor", input_tensor)); in TEST()
212 writer.WriteTensor(full_name("Tensor"), input_tensor).code()); in TEST()
H A Dserialization_utils.cc143 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteElementsToCheckpoint()
313 Status VariantTensorDataWriter::WriteTensor(StringPiece key, in WriteTensor() function in tensorflow::data::VariantTensorDataWriter
317 return WriteTensor(name, key, val); in WriteTensor()
320 Status VariantTensorDataWriter::WriteTensor(StringPiece name, StringPiece key, in WriteTensor() function in tensorflow::data::VariantTensorDataWriter
H A Dserialization_utils.h96 Status WriteTensor(StringPiece key, const Tensor& val) override;
97 Status WriteTensor(StringPiece name, StringPiece key,
H A Ddataset_utils.cc651 writer->WriteTensor(FullName(iterator_prefix, batch_prefix), in WriteBatch()
656 writer->WriteTensor(FullName(iterator_prefix, batch_prefix), in WriteBatch()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dsparse_tensor_slice_dataset_op.cc179 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
181 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
H A Dflat_map_dataset_op.cc234 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
H A Dwindow_dataset_op.cc282 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
H A Dinterleave_dataset_op.cc342 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveCurrentElements()
H A Dparallel_interleave_dataset_op.cc1288 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteElement()
1305 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteElement()
H A Dprefetch_dataset_op.cc258 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsummary_interface.h38 virtual Status WriteTensor(int64_t global_step, Tensor t, const string& tag,
H A Dsummary_kernels.cc150 OP_REQUIRES_OK(ctx, s->WriteTensor(step, *t, tag, serialized_metadata)); in Compute()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Drebatch_dataset_op.cc204 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
548 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
H A Dparallel_interleave_dataset_op.cc946 TF_RETURN_IF_ERROR(writer->WriteTensor(iterator_name, in WriteWorkerStateLocked()
1014 writer->WriteTensor(iterator_name, strings::StrCat(kInput, "_", i), in WriteWorkerThreadStateLocked()
1083 TF_RETURN_IF_ERROR(writer->WriteTensor( in WriteOutputElemLocked()
H A Dunique_dataset_op.cc124 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
H A Dunbatch_dataset_op.cc210 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
H A Dsliding_window_dataset_op.cc277 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
H A Dscan_dataset_op.cc281 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
H A Dgroup_by_reducer_dataset_op.cc321 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveInternal()
H A Dgroup_by_window_dataset_op.cc447 TF_RETURN_IF_ERROR(writer->WriteTensor( in SaveGroup()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Ddataset.h156 virtual Status WriteTensor(StringPiece key, const Tensor& val) = 0;
157 virtual Status WriteTensor(StringPiece name, StringPiece key,
/aosp_15_r20/external/tensorflow/tensorflow/core/util/tensor_bundle/
H A Dtensor_bundle.cc234 Status WriteTensor(const Tensor& val, FileOutputBuffer* out, in WriteTensor() function
465 status_ = WriteTensor(val, out_.get(), &data_bytes_written); in Add()

12