Searched refs:ImmutableTensor (Results 1 – 6 of 6) sorted by relevance
26 class ImmutableTensor {28 ImmutableTensor() = default;30 static ImmutableTensor Create(tensorflow::Tensor tensor);38 explicit ImmutableTensor(tensorflow::Tensor tensor) in ImmutableTensor() function58 explicit FallbackTensor(ImmutableTensor* immutable_tensor) in FallbackTensor()62 return absl::holds_alternative<ImmutableTensor*>(tensor_); in is_immutable()66 if (is_immutable()) return absl::get<ImmutableTensor*>(tensor_)->tensor(); in tensor()74 absl::variant<absl::monostate, tensorflow::Tensor, ImmutableTensor*> tensor_;
24 TEST(FallbackTensorTest, ImmutableTensor) { in TEST() argument28 auto immutable_tensor = ImmutableTensor::Create(tensor); in TEST()42 auto immutable_tensor = ImmutableTensor::Create(tensor); in TEST()66 auto immutable_tensor = ImmutableTensor::Create(tensor); in TEST()
61 ImmutableTensor ImmutableTensor::Create(tensorflow::Tensor tensor) { in Create()65 return ImmutableTensor( in Create()
39 int index, tensorflow::tfrt_stub::ImmutableTensor tensor) { in SetResource()46 resources_.push_back(std::make_unique<tensorflow::tfrt_stub::ImmutableTensor>( in SetResource()
43 void SetResource(int index, tensorflow::tfrt_stub::ImmutableTensor tensor);61 std::vector<std::unique_ptr<tensorflow::tfrt_stub::ImmutableTensor>>
668 tensorflow::tfrt_stub::ImmutableTensor::Create(arg.get().tensor())); in FallbackSetResource()912 tensorflow::tfrt_stub::ImmutableTensor::Create(tensor); in FallbackCopyTensorIfSmall()