Searched refs:TensorInterface (Results 1 – 5 of 5) sorted by relevance
78 tensorflow::TensorInterface ret( in CreateTensor()86 return new TF_Tensor{new tensorflow::TensorInterface(ret)}; in CreateTensor()175 tensorflow::down_cast<tensorflow::TensorInterface*>(to->tensor) in TF_TensorBitcastFrom()177 *tensorflow::down_cast<const tensorflow::TensorInterface*>( in TF_TensorBitcastFrom()185 void TensorInterface::Release() { delete this; } in Release()187 bool TensorInterface::CanMove() const { in CanMove()199 DataType TensorInterface::Type() const { return tensor_.dtype(); } in Type()201 int TensorInterface::NumDims() const { return tensor_.dims(); } in NumDims()203 int64_t TensorInterface::Dim(int dim_index) const { in Dim()207 int64_t TensorInterface::NumElements() const { in NumElements()[all …]
91 class TensorInterface : public AbstractTensorInterface {93 TensorInterface() {} in TensorInterface() function94 explicit TensorInterface(tensorflow::Tensor t) : tensor_(std::move(t)) {} in TensorInterface() function95 ~TensorInterface() override {} in ~TensorInterface()109 Status BitcastFrom(const TensorInterface& from, DataType type,119 return down_cast<TensorInterface*>(tensor)->Tensor(); in TensorFromInterface()
131 return new TensorInterface(Tensor(value)); in CreateInt64Scalar()135 return new TensorInterface(Tensor(value)); in CreateUint64Scalar()139 return new TensorInterface(Tensor(value)); in CreateInt32Scalar()143 return new TensorInterface(Tensor(value)); in CreateFloatScalar()147 return new TensorInterface(Tensor(value)); in CreateDoubleScalar()151 return new TensorInterface(Tensor(value)); in CreateHalfScalar()155 return new TensorInterface(Tensor(value)); in CreateStringScalar()160 return new TensorInterface(Tensor(value)); in CreateComplex128Scalar()164 return new TensorInterface(Tensor(value)); in CreateBoolScalar()169 return new TensorInterface(Tensor(dtype, TensorShape(dim_sizes))); in CreateTensor()
226 TensorInterface tensor_interface(std::move(tensor)); in TensorProtoToConstant()
789 tensorflow::TensorInterface interface(t); in TFE_OpSetAttrTensor()