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

  /external/tensorflow/tensorflow/compiler/xla/
shape_tree_test.cc 31 nested_tuple_shape_ = ShapeUtil::MakeTupleShape(
48 Shape nested_tuple_shape_; member in class:xla::__anon39023::ShapeTreeTest
82 TestShapeConstructor(nested_tuple_shape_, 10);
115 TestInitValueConstructor(nested_tuple_shape_, 10);
188 ShapeTree<int> shape_tree{nested_tuple_shape_};
196 EXPECT_TRUE(ShapeUtil::Compatible(nested_tuple_shape_, shape_tree.shape()));
229 ShapeTree<int> shape_tree{nested_tuple_shape_};
301 ShapeTree<int> destination(nested_tuple_shape_, 0);
320 ShapeTree<int> source(nested_tuple_shape_, 42);
372 ShapeTree<int> t(&nested_tuple_shape_, 42)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
copy_insertion_test.cc 555 nested_tuple_shape_, loop_state, 1));
602 nested_tuple_shape_, loop_state, 1));
689 nested_tuple_shape_, HloOpcode::kSelect, pred, tuple1, tuple2));
766 Shape nested_tuple_shape_ = member in class:xla::__anon38863::WhileCopyInsertionTest
    [all...]

Completed in 614 milliseconds