Searched defs:shape_tree (Results 1 – 6 of 6) sorted by relevance
121 ShapeTree<int> shape_tree{ShapeUtil::MakeTupleShape({})}; in TEST_F() local126 ShapeTree<int> shape_tree{array_shape_}; in TEST_F() local149 ShapeTree<int> shape_tree{tuple_shape_}; in TEST_F() local195 ShapeTree<int> shape_tree{nested_tuple_shape_}; in TEST_F() local230 ShapeTree<int> shape_tree{tuple_shape_}; in TEST_F() local237 ShapeTree<int> shape_tree{nested_tuple_shape_}; in TEST_F() local244 ShapeTree<std::unique_ptr<int>> shape_tree{tuple_shape_}; in TEST_F() local549 ShapeTree<int> shape_tree(shape); in BM_Construct() local564 ShapeTree<int> shape_tree(&shape); in BM_ConstructUnowned() local578 ShapeTree<int> shape_tree(shape); in BM_Copy() local[all …]
287 ShapeTree<bool> shape_tree(inst->shape(), true); in ComputeHloReplicationOnComputation() local293 ShapeTree<bool> shape_tree(inst->shape(), true); in ComputeHloReplicationOnComputation() local298 ShapeTree<bool> shape_tree(inst->shape(), false); in ComputeHloReplicationOnComputation() local312 ShapeTree<bool> shape_tree(inst->shape(), true); in ComputeHloReplicationOnComputation() local334 ShapeTree<bool> shape_tree(param->shape(), false); in ComputeHloReplication() local
87 ShapeTree<HloSharding> shape_tree = in TEST_F() local177 ShapeTree<HloSharding> shape_tree = in TEST_F() local226 ShapeTree<HloSharding> shape_tree(ShapeUtil::MakeTupleShape({}), in TEST_F() local234 ShapeTree<HloSharding> shape_tree(ShapeUtil::MakeTupleShape({}), in TEST_F() local
388 ShapeTree<HloSharding> shape_tree = GetAsShapeTree(shape); in ValidateTuple() local
259 const ShapeTree<llvm::Value*>& shape_tree = it->second; in ToString() local
256 XLA_MaybeOwningDeviceMemoryShapeTree shape_tree; member