Tensorflow security Rust/C++ interop

Bhack · April 21, 2021, 8:39pm

Do you think that Tensorflow could plan to introduce in 2021 Rust in the codebase like in other Google projects (Chromium, Android, Linux Kernel etc.) or do you think that we will have some specific issues for our library?

E.g. For Rust/C++ interop in Chromium see:

https://www.chromium.org/Home/chromium-security/memory-safety/rust-and-c-interoperability

/cc @mihaimaruseac

Bhack · April 21, 2021, 8:44pm

See also GitHub - google/autocxx: Tool for safe ergonomic Rust/C++ interop driven from existing C++ headers

mihaimaruseac · April 23, 2021, 10:56pm

Another related link: How CrowdStrike Combines TensorFlow and Rust for Performance

It would be nice to have parts of TF written in Rust instead of C++ for better safety, but I am unsure this is possible at the moment, unfortunately. We will need a large refactoring and modularization to allow this to happen.

Bhack · April 24, 2021, 9:24am

I think it could be interesting, as a prototipe, if we could consume the new WIP TF runtime c++ API (e.g. with autoccx) to write the simple kernel example in Rust:

github.com

tensorflow/runtime/blob/master/lib/test_kernels/simple_kernels.cc

// Copyright 2020 The TensorFlow Runtime Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// This file implements a few simple classes of synchronous kernels for testing.

#include <fstream>
#include <iterator>
#include <random>
#include <string>

This file has been truncated. show original

Bhack · June 17, 2021, 11:08pm

It could be really nice to have an analysis like this one for TF