Searched defs:reshaped_dims (Results 1 – 3 of 3) sorted by relevance
985 std::vector<int64_t> reshaped_dims( in PartitionBaseCase() local996 std::vector<int64_t> reshaped_dims( in PartitionBaseCase() local1090 std::vector<int64_t> reshaped_dims( in PartitionBaseCase() local1100 std::vector<int64_t> reshaped_dims( in PartitionBaseCase() local1242 std::vector<int64_t> reshaped_dims( in PartitionBaseCase() local1263 std::vector<int64_t> reshaped_dims( in PartitionBaseCase() local1373 std::vector<int64_t> reshaped_dims( in PartitionBaseCase() local
461 std::vector<int64_t> reshaped_dims(warp_dims_without_last_dims.size()); in CalculateGradWarp() local
331 DSizes<Index, ReshapedDims> reshaped_dims; in test_execute_reshape() local