HomeSort by relevance Sort by last modified time
    Searched refs:ReshapeTest (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
reshape_test.cc 45 class ReshapeTest : public ::testing::WithParamInterface<bool>,
48 ReshapeTest() { set_use_bfloat16(GetParam()); }
54 XLA_TEST_P(ReshapeTest, CollapseTrivial1x1) {
69 XLA_TEST_P(ReshapeTest, CollapseTrivialR1EmptyDims) {
82 XLA_TEST_P(ReshapeTest, CollapseTrivialR1OnlyDim) {
96 XLA_TEST_P(ReshapeTest, SingleElementArrayToScalar) {
113 XLA_TEST_P(ReshapeTest, ScalarToSingleElementArray) {
132 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(Trivial0x3)) {
148 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(Trivial0x3WithParameter)) {
165 XLA_TEST_P(ReshapeTest, DISABLED_ON_GPU(Trivial3x0))
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
reshape_op_test.py 30 class ReshapeTest(test.TestCase):
  /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
ops_test.py 239 class ReshapeTest(Base):
    [all...]

Completed in 188 milliseconds