Home
last modified time | relevance | path

Searched refs:WriteBinaryProto (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/
H A Dxplane_to_hlo.cc67 TF_RETURN_IF_ERROR(tensorflow::WriteBinaryProto(tensorflow::Env::Default(), in GetHloProtoFromMultiXSpaceAndSaveToFile()
81 TF_RETURN_IF_ERROR(tensorflow::WriteBinaryProto( in GetHloProtoFromMultiXSpaceAndSaveToFile()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/inputs/
H A Dutils_test.cc47 TF_CHECK_OK(WriteBinaryProto(env_, binary_graph_def_file_, graph_def_)); in SetUp()
52 WriteBinaryProto(env_, binary_meta_graph_def_file_, meta_graph_def_)); in SetUp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/
H A Dconvert_computation.cc41 TF_CHECK_OK(tensorflow::WriteBinaryProto(env, path, module)); in RealMain()
H A Drun_hlo_module.cc71 tensorflow::WriteBinaryProto(env, binary_filename, literal.ToProto())); in WriteLiteralToTempFile()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dprotobuf_util.cc64 return tensorflow::WriteBinaryProto(env, *full_path, message); in DumpProtoToDirectory()
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfile_utils_test.cc50 TF_ASSERT_OK(WriteBinaryProto(Env::Default(), binary_file, graph_def)); in TestLoadTextOrBinaryGraphFile()
H A Dtransform_graph.cc273 save_status = WriteBinaryProto(Env::Default(), out_graph, graph_def); in ParseFlagsAndTransformGraph()
/aosp_15_r20/external/tensorflow/tensorflow/tools/tfg_graph_transforms/
H A Dutils.h88 tensorflow::WriteBinaryProto(tensorflow::Env::Default(), output_file, in SerializeProto()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dliteral_test_util.cc41 TF_CHECK_OK(tensorflow::WriteBinaryProto(env, absl::StrCat(filename, ".pb"), in WriteLiteralToTempFile()
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/rpc/
H A Dprofiler_service_impl.cc68 return WriteBinaryProto(Env::Default(), out_path, xspace); in CollectDataToRepository()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/tests/
H A Dxla_compilation_cache_test_helper.cc155 TF_RETURN_IF_ERROR(WriteBinaryProto(env, file_path, entry)); in AlterPersistentCacheEntryHloModuleNames()
/aosp_15_r20/external/tensorflow/tensorflow/core/platform/
H A Denv_test.cc121 TF_EXPECT_OK(WriteBinaryProto(env_, filename, proto)); in TEST_F()
H A Denv.h624 Status WriteBinaryProto(Env* env, const std::string& fname,
H A Denv.cc585 Status WriteBinaryProto(Env* env, const string& fname, in WriteBinaryProto() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddump.cc641 tensorflow::WriteBinaryProto(env, absl::StrCat(path, ".pb"), proto); in DumpProtobufToFile()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_compilation_cache.cc894 return WriteBinaryProto(env, file_path, entry); in SaveSerializedEntry()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dsnapshot_utils.cc951 TF_RETURN_IF_ERROR(WriteBinaryProto(env, tmp_filename, *metadata)); in WriteMetadataFile()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dmodel.cc2765 return WriteBinaryProto(Env::Default(), fname, model_proto); in Save()