mirror of
https://github.com/twitter/the-algorithm.git
synced 2025-06-08 05:39:03 -05:00
ef4c5eb65e
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
14 lines
458 B
C++
14 lines
458 B
C++
#pragma once
|
|
|
|
#include "tensorflow/core/framework/op.h"
|
|
#include "tensorflow/core/framework/op_kernel.h"
|
|
#include "tensorflow/core/framework/shape_inference.h"
|
|
#include <twml.h>
|
|
|
|
using namespace tensorflow;
|
|
twml::Tensor TFTensor_to_twml_tensor(Tensor &input);
|
|
twml::RawTensor TFTensor_to_twml_raw_tensor(Tensor &input);
|
|
const twml::Tensor TFTensor_to_twml_tensor(const Tensor &input);
|
|
const twml::RawTensor TFTensor_to_twml_raw_tensor(const Tensor &input);
|
|
|