Searched refs:immutable_tensor (Results 1 – 3 of 3) sorted by relevance
28 auto immutable_tensor = ImmutableTensor::Create(tensor); in TEST() local30 ASSERT_EQ(immutable_tensor.tensor().NumElements(), 1); in TEST()31 ASSERT_EQ(immutable_tensor.tensor().dtype(), tensorflow::DT_INT32); in TEST()32 auto flat = immutable_tensor.tensor().flat<int32_t>(); in TEST()34 EXPECT_FALSE(immutable_tensor.tensor().RefCountIsOne()); in TEST()35 EXPECT_EQ(tensor.TotalBytes(), immutable_tensor.tensor().TotalBytes()); in TEST()42 auto immutable_tensor = ImmutableTensor::Create(tensor); in TEST() local44 ASSERT_EQ(immutable_tensor.tensor().NumElements(), 1); in TEST()45 ASSERT_EQ(immutable_tensor.tensor().dtype(), tensorflow::DT_STRING); in TEST()46 auto flat = immutable_tensor.tensor().flat<tensorflow::tstring>(); in TEST()[all …]
58 explicit FallbackTensor(ImmutableTensor* immutable_tensor) in FallbackTensor() argument59 : tensor_(immutable_tensor) {} in FallbackTensor()
911 auto immutable_tensor = in FallbackCopyTensorIfSmall() local915 std::move(immutable_tensor.tensor())); in FallbackCopyTensorIfSmall()