Searched refs:kMaxTensorSize (Results 1 – 3 of 3) sorted by relevance
282 static constexpr int64 kMaxTensorSize = 1024; variable
61 static constexpr int64 kMaxTensorSize = ShapeRefiner::kMaxTensorSize; member in tensorflow::ShapeRefinerTest265 root, {static_cast<int32>(2), TensorShape({4, kMaxTensorSize / 2})}); in TEST_F()295 EXPECT_EQ(strings::StrCat("[", kMaxTensorSize * 2 * 3, "]"), in TEST_F()
465 if (outputs[0].TotalBytes() <= kMaxTensorSize) { in EvaluateConstantTensorForEdge()