Searched defs:GetInputSize (Results 1 – 6 of 6) sorted by relevance
36 int64_t GetInputSize() override { in GetInputSize() function in xla::exhaustive_op_test::ExhaustiveF64UnaryTest
37 int64_t GetInputSize() override { in GetInputSize() function in xla::exhaustive_op_test::__anoncffb54ac0111::Exhaustive16BitBinaryTest
83 int64_t GetInputSize() override { in GetInputSize() function in xla::exhaustive_op_test::ExhaustiveComplexUnaryTestBase
40 int64_t GetInputSize() override { in GetInputSize() function in xla::exhaustive_op_test::__anonc34c31e10111::Exhaustive32BitOrMoreBinaryTest
187 int64_t GetInputSize() override { in GetInputSize() function in xla::exhaustive_op_test::Exhaustive32BitOrLessUnaryTest
557 unsigned int GetInputSize(unsigned int inputIndex = 0u) const in GetInputSize() function in InferenceModel