Searched defs:GetInputSize (Results 1 – 6 of 6) sorted by relevance
36 int64 GetInputSize() override { in GetInputSize() function in xla::exhaustive_op_test::ExhaustiveF64UnaryTest
37 int64 GetInputSize() override { in GetInputSize() function in xla::exhaustive_op_test::__anonbafe06220111::Exhaustive16BitBinaryTest
83 int64 GetInputSize() override { in GetInputSize() function in xla::exhaustive_op_test::ExhaustiveComplexUnaryTestBase
40 int64 GetInputSize() override { in GetInputSize() function in xla::exhaustive_op_test::__anon0ea512170111::Exhaustive32BitOrMoreBinaryTest
184 int64 GetInputSize() override { in GetInputSize() function in xla::exhaustive_op_test::Exhaustive32BitOrLessUnaryTest
486 unsigned int GetInputSize(unsigned int inputIndex = 0u) const in GetInputSize() function in InferenceModel