Searched defs:shape_tree (Results 1 – 6 of 6) sorted by relevance
122 ShapeTree<int> shape_tree{ShapeUtil::MakeTupleShape({})}; in TEST_F() local129 ShapeTree<int> shape_tree{shape}; in TEST_F() local134 ShapeTree<int> shape_tree{array_shape_}; in TEST_F() local157 ShapeTree<int> shape_tree{tuple_shape_}; in TEST_F() local203 ShapeTree<int> shape_tree{nested_tuple_shape_}; in TEST_F() local238 ShapeTree<int> shape_tree{tuple_shape_}; in TEST_F() local245 ShapeTree<int> shape_tree{nested_tuple_shape_}; in TEST_F() local252 ShapeTree<std::unique_ptr<int>> shape_tree{tuple_shape_}; in TEST_F() local562 ShapeTree<int> shape_tree(shape); in BM_Construct() local577 ShapeTree<int> shape_tree(&shape); in BM_ConstructUnowned() local[all …]
347 ShapeTree<HloReplication> shape_tree( in ComputeHloReplicationOnComputation() local354 ShapeTree<HloReplication> shape_tree( in ComputeHloReplicationOnComputation() local360 ShapeTree<HloReplication> shape_tree( in ComputeHloReplicationOnComputation() local379 ShapeTree<HloReplication> shape_tree( in ComputeHloReplicationOnComputation() local403 ShapeTree<HloReplication> shape_tree(param->shape(), in ComputeHloReplication() local
88 ShapeTree<HloSharding> shape_tree = in TEST_F() local177 ShapeTree<HloSharding> shape_tree = in TEST_F() local233 ShapeTree<HloSharding> shape_tree(ShapeUtil::MakeTupleShape({}), in TEST_F() local241 ShapeTree<HloSharding> shape_tree(ShapeUtil::MakeTupleShape({}), in TEST_F() local
558 ShapeTree<HloSharding> shape_tree = GetAsShapeTree(shape); in ValidateTuple() local
230 const ShapeTree<llvm::Value*>& shape_tree = it->second; in ToString() local
283 XLA_MaybeOwningDeviceMemoryShapeTree shape_tree; member